2016校招真题之二维数组打印

1、题目描述

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

2、代码实现

package com.wcy.october;

/**
 * 时间：2016年7月18日
 * 题目：二维数组打印
 * 题目描述：有一个二维数组(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 class Printer {

    /**
     * 从右上角到左下角沿主对角线方向打印数组
     * @param arr
     * @param n
     * @return
     */
    public int[] arrayPrint(int[][] arr, int n) {
        int[] resultArray = new int[n*n];
        int num = 0;
        for (int i = n-1; i >= 0; i--) {
            for (int j = 0; j < n-i; j++) {
                resultArray[num] = arr[j][i+j];
                resultArray[n*n-1-num] = arr[n-1-j][n-1-i-j];
                num++;
            }
        }
        return resultArray;
    }
    
    /**
     * 打印数组函数
     * @param arr 待打印的数组
     */
    public void showArray(int[] arr){
        System.out.print("[");
        for (int i = 0; i < arr.length; i++) {
            if (i == arr.length-1) {
                System.out.print(arr[i]);
            }else {
                System.out.print(arr[i] + ",");
            }
        }
        System.out.println("]");
    }
    
    /**
     * 用户页面测试
     * @param args
     */
    public static void main(String[] args) {
        Printer test = new Printer();
        int[][] arr = {{1,2,3,4},{5,6,7,8},{9,10,11,12},{13,14,15,16}};
        int[] resultArray = test.arrayPrint(arr, arr.length);
        test.showArray(resultArray);
    }
    
}