this not a good solution but good enough for the test cases
Distance walked 1D
Sum of diagonal of a square matrix
Getting the indices from a vector
Product of elements in row
Find the treasures in MATLAB Central and discover how the community can help you!
Choose a web site to get translated content where available and see local events and offers. Based on your location, we recommend that you select: .
You can also select a web site from the following list:
Select the China site (in Chinese or English) for best site performance. Other MathWorks country sites are not optimized for visits from your location.
Contact your local office