蛇形填数（c++）

如下图所示，小明用从 1 开始的正整数“蛇形”填充无限大的矩阵。

1 2 6 7 15 ...
3 5 8 14 ...
4 9 13 ...
10 12 ...
11 ...
...

容易看出矩阵第二行第二列中的数是 5。请你计算矩阵中第 20 行第 20 列的数是多少？

解题思路：

找到如下规律

然后按以下坐标依次按顺序赋值

11

12 21 (x坐标递增，y坐标递减）

31 22 13 (x坐标递减，y坐标递增）

14 23 32 41..........（x坐标递增，y坐标递减）以此类推

#include <iostream>
using namespace std;
int arr[1000][1000];
int main()
{
  int flag = 0;
  int num=1;
  for(int i = 1;i<=50;i++){
    if(flag==1){ //flag=1时，(x坐标递增，y坐标递减）
      for(int j = 1,k=i;j<=i;j++,k--){
        // cout << j << "/" << k << " ";
        arr[j][k]=num;
        num++;
      }
      flag=0;
    }else{ //flag=0时， (x坐标递减，y坐标递增）
      for(int j = i,k=1;j>0;j--,k++){
        // cout << j << "/" << k << " ";
        arr[j][k]=num;
        num++;
      }
      flag=1;
    }
    
  }
  cout << arr[20][20];
  return 0;
  // 11 
  // 12 21 
  // 31 22 13 
  // 14 23 32 41
}