美团点评2016研发题目-[编程题-美团] 二维数组打印

[编程题] 二维数组打印

有一个二维数组(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；注意暂存横纵坐标
*
* @author mc
*
*/
public class Main {
public int[] arrayPrint(int[][] arr, int n) {
int[] res = new int[n * n];
int idx = 0;
// 右上角（包括主对角线）
for (int k = n - 1; k >= 0; k--) {
int i = 0;
int j = k;
while (i < n && j < n) {
res[idx] = arr[i][j];
i++;
j++;
}
}
// 左下角（不包括主对角线,行从1开始）
for (int k = 1; k < n; k++) {
int i = k;
int j = 0;
while (i < n && j < n) {
res[idx] = arr[i][j];
i++;
j++;
idx++;
}
}
return res;
}
}