二维数组打印

 

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

分析：沿着主对角线方向，每次元素下标都要各加1，直到超出边界；右上角的每次边界后都是列减一，左下角每次边界后都是行加1，对于这个边界条件就是行行有没有到n。

vector<int> arrayPrint(vector<vector<int> > arr, int n) {
            // write code here
            //从第一行最后一列开始
            int row=0;
            int col=n-1;
            
            vector<int> vec;
            
            while(row<n)//行完了就是完了
            {
                //i和j是遍历每一个符合条件元素，row和cow记录本次起始行。列
                int i=row;
                int j=col;
                //因为i和j所代表的行列都要加加，因此下标合法性判断都是小于n
                while(i<n&&j<n)
                {
                    //主对角线每次都是行列均加1，不合法时再调整行列
                    vec.push_back(arr[i][j]);
                    i++;
                    j++;
                }
                if(j==n&&i<n)//当列超出时(并且行合法)，看前一列(行不变)
                    col--;
                if(i==n)//当行超出时，即主对角线那条完了，则看下一行(列不变)
                    row++;  
            }
            return vec;
        }