美团 二维数组打印

题目描述

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

public int[] arrayPrint(int[][] arr, int n) { int[] res = new int[n * n]; int index = 0; for (int y = n - 1; y > 0; y--) { int i = 0; for (int j = y; j < n;) { res[index++] = arr[i++][j++]; } } for (int x = 0; x < n; x++) { int j = 0; for (int i = x; i < n; ) { res[index++] = arr[i++][j++]; } } return res; }