【美团】二维数组打印

题目描述

有一个二维数组(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) {
        // write code here
        vector<int> ans;
        for(int j = n-1;j>-1;j--)
            for (int i = 0;i<n-j;i++)
                ans.push_back(arr[i][j+i]);
        
        for(int i = 1;i<n;i++)
            for(int j = 0;j<n-i;j++)
                ans.push_back(arr[i+j][j]);
        
        return ans;
    }
};