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

题目描述

[[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;
}
}