蓝桥杯蛇形填数

905853663

【问题描述】
如下图所示，小明用从 1 开始的正整数“蛇形”填充无限大的矩阵。
1 2 6 7 15 …
3 5 8 14 …
4 9 13 …
10 12 …
11 …
…
容易看出矩阵第二行第二列中的数是 5。请你计算矩阵中第 20 行第 20 列
的数是多少？

代码如下：
public class Main{

        public static void main(String[] args) {
                int[][] a = new int[60][60];
                int x = 1,i = 0,j = 0;
                for (int j2 = 0; j2 < a.length; j2++) {
                        for (int k = 0; k < a.length; k++) {
                                a[j2][k] = 0;
                        }
                }
                while (i<60&&j<60) {
                        if (i==0&&j%2==0) {
                                a[i][j] = x++;
                                a[i][j++] = x++;
                                while (j>0) {
                                        a[i++][j--] = x++;
                                }
                        }
                        if (i%2==1&&j==0) {
                                a[i][j] = x++;
                                a[i++][j] = x++;
                                while (i>=0) {
                                        a[i--][j++] = x++;
                                }
                        }
                }
                System.out.println(a[19][19]);

        }
}




输不出结果，而且编译器都卡了，这个时间复杂度不算高把？请问有大佬知道我的错误在哪吗？

wp231957

还用这么复杂吗，用公式就能算出来
1  5  13   25   41    61   85   113   145
181     221   265    313   365   421    481
545    613    685   761

905853663

wp231957 发表于 2022-2-26 07:10
还用这么复杂吗，用公式就能算出来
1  5  13   25   41    61   85   113   145
181     221   265    31 ...

我知道，但是还是想研究一下

傻眼貓咪

答案是 761 吗？
Main.java

public class Main {
    public static void main(String[] args) {
        int N = 20;
        // int[][] arr = new int[N*2+1][N*2+1]; 因为题目只要求矩阵第 20 行第 20 列的值，所以不用全部值都储存
        boolean flag = true;
        for(int num = 1, a, b = a = 0;; num++) { // a = 0 为上边界，b = 0 为左边界，这里没有设右边界和下边界，因为不需要。
            // arr[a][b] = num; 如上，题目只要求矩阵第 20 行第 20 列的值，不必储存
            if(a == N-1 && b == N-1){
                System.out.println(num); // 
                break;
            }
            if(flag){ // flag = true 往下
                if(b == 0){ // 抵达边界，则反转
                    flag = !flag;
                }
                else b--;
                a++;
            }
            else{ // flag = false 往上
                if(a == 0){ // 抵达边界，则反转
                    flag = !flag;
                }
                else a--;
                b++;
            }
        }
    }
}

761

905853663

傻眼貓咪 发表于 2022-2-26 15:39
答案是 761 吗？
Main.java

是761，但是我还是想知道我的那个代码错在哪里l