题目描述
有一个二维数组(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; } }