问题 E: 素数方阵（C++算法题）

时间限制: 1.000 Sec  内存限制: 128 MB

题目描述

把前n2个素数从左上角开始按右、下、左、上、右、下、左、上……的顺序填入n×n的方阵就得到了蛇形素数方阵。以下是n=4和n=5的蛇形素数方阵：

给出n，你的任务是求出n×n的蛇形素数方阵，并输出其中某个方格中的数值。
素数，又称质数，是指除1和其自身之外，没有其他约数的大于1的正整数。

输入

输入一行三个正整数n,x,y(1≤x,y≤n)。

输出

输出一行一个整数，表示n × n蛇形素数方阵第x行第y列中的数字。

样例输入 Copy

【样例1】
5 1 4
【样例2】
5 4 3

样例输出 Copy

【样例1】
7
【样例2】
79

提示

所有数据满足1 ≤ n ≤ 20。

答案如下

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;

bool is_prime(int n){
    for (int i=2;i<=n/i;i++){
        if(n%i==0) return false;
    }
    return true;
}

vector<ll> Prime(int n){
    vector<ll>B;
        for (int i=2,s=0;s<n*n;i++){
            if(is_prime(i)){
                B.push_back(i);
                s++;
            }
        }
        return B;
}
ll C[21][21]={0};
int main(){
    int n,x,y;
    cin>>n>>x>>y;
    auto B=Prime(n);

    int k = 0,q = 0;
    while (n - 2*q > 1){
    for (int i = 0 + q; i < n - q; i++) {
    }
    for (int i = 1 + q; i < n - q; i++) {
        C[i][n - 1 - q] = B[k++];
    }
    for (int i = n - 2 - q; i >= 0+q; i--) {
        C[n - 1 - q][i] = B[k++];
    }
    for (int i = n - 2 - q; i > 0+q; i--) {
        C[i][q] = B[k++];
    }
    q++;
}
C[(n+1)/2][(n+1)/2]=B[k];
cout << C[x-1][y-1] << endl;
}

这种蛇皮走位的数组在循环I和循环II专题里经常出现，可以去复习一下。
利用好i与n之间的关系，变量能少则少。