luogu P1579 哥德巴赫猜想（升级版）

题目描述

一个等差数列是一个能表示成a, a+b, a+2b,..., a+nb (n=0,1,2,3,...)的数列。

在这个问题中a是一个非负的整数，b是正整数。写一个程序来找出在双平方数集合(双平方数集合是所有能表示成p的平方 + q的平方的数的集合,其中p和q为非负整数)S中长度为n的等差数列。

输入输出格式

输入格式：

第一行: N(3<= N<=25),要找的等差数列的长度。

第二行: M(1<= M<=250),搜索双平方数的上界0 <= p,q <= M。

输出格式：

如果没有找到数列,输出`NONE‘。

如果找到了，输出一行或多行, 每行由二个整数组成:a,b。

这些行应该先按b排序再按a排序。

所求的等差数列将不会多于10,000个。

输入输出样例

输入样例#1：

5
7

输出样例#1：

1 4
37 4
2 8
29 8
1 12
5 12
13 12
17 12
5 20
2 24

说明

题目翻译来自NOCOW。

USACO Training Section 1.

两个点re但本地不re为什么？

#include<cstdio>
#include<cmath>
using namespace std;
int n;
bool a[400006];
int num_prime=0, prime[400006];
void init()
{
    scanf("%d",&n);
    a[0]=a[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(!a[i])
            prime[num_prime++]=i;
        for(int j=0;j<num_prime&&i*prime[j]<=n;j++)
        {
            a[i*prime[j]]=1;
            if(!(i%prime[j]))
                break;
        }
    }
}

void solve(int x)
{
    int i=0,j=0;
    while(prime[i]<x)
    {
        int tmp=x-prime[i];
        j=0;
        while(prime[j]<tmp)
        {
            int ttmp=tmp-prime[j];
            if(!a[ttmp])
            {
                printf("%d %d %d\n",prime[i],prime[j],ttmp);
                return;
            }
            j++;
        }
        i++;
    }
}
int main()
{
    init();
    solve(n);
    return 0;
}