Cody

Problem 2587. Find indices of diagonal elements

Solution 2738338

Submitted on 24 Jul 2020 by Rafael S.T. Vieira
This solution is locked. To view this solution, you need to provide a solution of the same size or smaller.

Test Suite

Test Status Code Input and Output
1   Pass
x = [1 2 3;4 5 6;7 8 9]; a = [1 5 9]; assert(isequal(diag_indices(x,0),a))

2   Pass
x = magic(5); b = 5; assert(isequal(diag_indices(x,-4),b))

3   Pass
x = magic(5); a = 21; assert(isequal(diag_indices(x,4),a))

4   Pass
x = magic(5); z = 5; assert(isequal(diag_indices(x,-4),z))

5   Pass
x = magic(6); a = [19 26 33]; assert(isequal(diag_indices(x,3),a))

6   Pass
x1 = magic(50); y_correct = 50; assert(isequal(diag_indices(x1,-49),y_correct))

7   Pass
x2 = magic(50); y_correct = [49 100]; assert(isequal(diag_indices(x2,-48),y_correct))

8   Pass
y = magic(50); v = [48 99 150]; assert(isequal(diag_indices(y,-47),v))

9   Pass
v=[1 2 3;4 5 6;1 2 3]; t = [1 5 9]; assert(isequal(diag_indices(v,0),t))

10   Pass
x=[1 2;1 2]; y_correct = 3; assert(isequal(diag_indices(x,1),y_correct))

11   Pass
x=[1 1 1;1 1 1;1 1 1]; t = [1 5 9]; assert(isequal(diag_indices(x,0),t))

12   Pass
y=[1 2 3;1 2 3]; y_correct = [1 4]; assert(isequal(diag_indices(y,0),y_correct))