华为机试---二维数组打印



题目描述

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

import java.util.*;

public class Printer {
    public int[] arrayPrint(int[][] arr, int n) {
        // 从右上角到左下角沿主对角线方向打印
        int[] res = new int[n * n];
        int count = 0;
        //先打印右边主对角线以上三角，包括主对角线
        //列减 行增
        for (int width = n - 1, hight = 0; width >= 0 && hight <= n - 1; width--, hight++) {  //控制上边坐标和右边坐标
            for (int i = width, j = 0; i <= n - 1 && j <= hight; i++, j++) { //斜着打印             
                res[count++] = arr[j][i];}
        }
        //打印左下主对角线以下三角
        for (int hight = 1, width = n - 2; hight <= n - 1 && width >= 0; hight++, width--) {  //控制左边左边和下边坐标
            for (int i = hight, j = 0; i <= n - 1 && j <= width; i++, j++) { //斜着打印  横坐标和纵坐标都相加             
                res[count++] = arr[i][j];
            }
        }
        return res;
    }
}