8 、Acwing 2021/1/20 756.蛇形矩阵

8 、Acwing 2021/1/20 756.蛇形矩阵

756. 蛇形矩阵

输入两个整数n和m，输出一个n行m列的矩阵，将数字 1 到 n*m 按照回字蛇形填充至矩阵中。

具体矩阵形式可参考样例。

输入格式

输入共一行，包含两个整数n和m。

输出格式

输出满足要求的矩阵。

矩阵占n行，每行包含m个空格隔开的整数。

数据范围

1≤n,m≤1001≤n,m≤100

输入样例：

3 3

输出样例：

1 2 3
8 9 4
7 6 5

import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int[][] q = new int[n][m]; //因为最大的数据是100，多加10防止溢出
        
        //四个方向,上下左右
        int[] dx = {-1 , 0 , 1 , 0};
        int[] dy = {0 , 1 , 0 , -1};
        
        int x = 0 , y = 0;
        int d = 1; //用于改变方向
        
        for(int i = 1 ; i <= n * m ; i ++){
            q[x][y] = i ; //填入数
            
            //方向
            int a = x + dx[d];
            int b = y + dy[d];
            
            //判断边界
            if(a < 0 || a >= n || b < 0 || b >= m || q[a][b] != 0){
                d = (d + 1) % 4; //技巧，当d等于4时又变为 0
                
                //更新
                a = x + dx[d];
                b = y + dy[d];
            }
            
            x = a;
            y = b;
        }
        for(int i = 0 ; i < n ; i ++){
            for(int j = 0 ; j < m ; j++){
                System.out.print(q[i][j]+" ");
            }
            System.out.println();
        }
        
    }
}