2899: 蛇形填充数组

解题思路:

注意事项:

参考代码:

import java.util.Scanner;

public class Main {
   public static void main(String[] args) {
       Scanner scanner = new Scanner(System.in);
       int n = scanner.nextInt();
       int[][] matrix = constructSnakeMatrix(n);
       printSnakeMatrix(matrix);
   }

   public static int[][] constructSnakeMatrix(int n) {
       int[][] matrix = new int[n][n];

       // 初始化起始位置
       int row = 0;
       int col = 0;

       // 填充方阵
       for (int num = 1; num <= n*n; num++) {
           matrix[row][col] = num; // 将数字写入当前位置

           // 下一个数字的位置计算
           if ((row + col) % 2 == 0) { // 编号为奇数的斜线
               if (col == n - 1) { // 达到行末时向下移动一行
                   row++;
               } else if (row == 0) { // 达到列首时向右移动一列
                   col++;
               } else { // 向上移动一行，向右移动一列
                   row--;
                   col++;
               }
           } else { // 编号为偶数的斜线
               if (row == n - 1) { // 达到列末时向右移动一列
                   col++;
               } else if (col == 0) { // 达到行首时向下移动一行
                   row++;
               } else { // 向下移动一行，向左移动一列
                   row++;
                   col--;
               }
           }
       }

       return matrix;
   }

   public static void printSnakeMatrix(int[][] matrix) {
       for (int[] row : matrix) {
           for (int num : row) {
               System.out.print(num + " ");
           }
           System.out.println();
       }
   }
}