c++-二维数组打印对角线打印

题目描述

有一个二维数组(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循环是对左下角进行处理