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

来源:互联网 发布:js 弹出提示框 编辑:程序博客网 时间:2024/05/19 14:17


题目描述

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