美团点评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;
	}
}