二维数组打印----美团2016研发工程师编程题(二)

[编程题] 二维数组打印

有一个二维数组(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]

class Printer {
public:
    vector<int> arrayPrint(vector<vector<int> > arr, int n) {
        int row=0,col=n-1;
        vector<int> vec;
        while(row<n){
            int i=row,j=col;
            while(i<n&&j<n){
                vec.push_back(arr[i][j]);
                i++;
                j++;
            }
            if(j==n&&i<n)
                col--;
            if(i==n)
                row++;  
        }
        return vec;
    }
};