题目描述
有一个二维数组(n*n),写程序实现从右上角到左下角沿主对角线方向打印。
给定一个二位数组arr及题目中的参数n,请返回结果数组。
测试样例:
[[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]],4
返回:[4,3,8,2,7,12,1,6,11,16,5,10,15,9,14,13]
题目很容易理解,思路就是利用对角线中,相邻成员的下标递增,a[i][i]的下一个就是a[i+1][i+1],利用好这一点即可。
然后分右上角和左下角两部分进行处理:
class Printer {
public:
vector<int> arrayPrint(vector<vector<int> > arr, int n) {
// write code here
vector<int> res;
for (int i=n-1;i>=0;i--){
int x=0;
int y=i;
while(x<n && y<n){
res.push_back(arr[x][y]);
x++;
y++;
}
}
for(int j=1;j<n;j++) {
int x=j;
int y=0;
while(x<n && y<n){
res.push_back(arr[x][y]);
x++;
y++;
}
}
return res;
}
};
第一个for循环是对右上角进行处理,第二个for循环是对左下角进行处理