Codeforces B. Prime Square (#678 Div.2) (思维构造 / 素数与和数)

传送门

题意: 需要你构造一个n*n的矩阵,满足其下三个条件:

  • 每个数都在1e5以内
  • 每个数都不是素数
  • 每一行或列的和都是个素数

在这里插入图片描述
思路:

  • 刚开始一直在想简便的特殊方法,结果一直wa1,最后才想到数据范围这么小应该可以暴力尝试过了(′д` )…彡…彡。
  • 我们只需要找n个数满足其和是素数且每个数不是素数即可。
  • 那么我们让其中n-1个都是1,只需要再找到一个和数 x 满足 x+n-1 是个素数即可。
  • 后期构造的时候让x后移轮换即可。

代码实现:

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {
    
    {
    
    1, 0}, {
    
    -1, 0}, {
    
    0, 1}, {
    
    0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 1e5 + 5;

int t, n;

int check(int x){
    
    
    for(int i = 2; i < x; i ++)
        if(x%i==0) return 0;
    return 1;
}

signed main()
{
    
    
    IOS;

    cin >> t;
    while(t --){
    
    
        cin >> n;

        int x;
        for(int i = n; i < N; i ++){
    
    
            if(check(i) && !check(i-n+1)) {
    
    x = i-n+1; break;}
        }

        for(int i = 1; i <= n; i ++){
    
    
            for(int j = 1; j <= n; j ++){
    
    
                if(i == j) cout << x << " \n"[j==n];
                else cout << 1 << " \n"[j==n];
            }
        }
    }

    return 0;
}

猜你喜欢

转载自blog.csdn.net/Satur9/article/details/109302765
今日推荐