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循环是对左下角进行处理