Z型打印二维数组(模拟)

Posted 2021-10-07 nuist__NJUPT

Z型打印二维数组

样例输入：
3 4
1 2 3 4
5 6 7 8
9 10 11 12

样例输出：
1 2 5 9 6 3 4 7 10 11 8 12

算法思想：本题打印数组分为从左往右和从右往左，若为从左往右，则考虑三种情况，若是从右往左，则也考虑三种情况。

import java.util.Scanner;

public class PrintArray {
    public static void print(int [][] matrix){
        int r = 0, c = 0 ;
        int m = matrix.length, n = matrix[0].length ;
        boolean leftToRight = true ; //标记从左到右
        while(r < m && c < n){
            if(leftToRight){ //从左往右走
                System.out.print(matrix[r][c] + " ") ;
                if(r == 0 && c < n - 1){
                    leftToRight = !leftToRight ;
                    c ++ ;
                    continue ;
                }else if(r > 0 && c == n - 1){
                    leftToRight = !leftToRight ;
                    r ++ ;
                    continue ;
                }else{
                    r -- ;
                    c ++ ;
                }
            }else{ //从右往左走
                System.out.print(matrix[r][c] + " ") ;
                if(c == 0 && r < n - 1){
                    leftToRight = !leftToRight ;
                    r ++ ;
                    continue ;
                }else if(r == m -1 && c < n-1){
                    leftToRight = !leftToRight ;
                    c ++ ;
                    continue ;
                }else{
                    r ++ ;
                    c -- ;
                }
            }
        }
    }
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in) ;
        int m = input.nextInt() ;
        int n = input.nextInt() ;
        int [][] matrix = new int [m][n] ;
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                matrix[i][j] = input.nextInt() ;
            }
        }
        print(matrix) ;
    }
}