USACO1.4.1 Arithmetic Progressions (ariprog)

时限为5s

所以先把双平方数集合预处理出来，暴力搜索枚举即可，注意及时剪枝

输出答案，应该先按b排序再按a排序

/*
ID:xsy97051
LANG:C++
TASK:ariprog
*/
#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;

int rl=0,a[20000],b[20000];
int n,m,num[30000],len=0;
bool vis[125001];

void init(int m)//预处理双平方数集合
{
	for(int i=0;i<=125000;i++)
        vis[i]=0;
	
    for(int i=0;i<=m;i++)
        for(int j=0;j<=m;j++)
		    vis[i*i+j*j]=1;
	
	for(int i=0;i<=m*m*2;i++)
		if(vis[i]) 
		{
			len++;
            num[len]=i;
    	}
    return;
}   

int main()
{
    freopen("ariprog.in","r",stdin);
	freopen("ariprog.out","w",stdout);
	cin>>n>>m;
    init(m);
    m=m*m*2;
    
    for(int i=1;i<=len;i++)
        for(int j=i+1;j<=len;j++)
        {
            if((2-n)*num[i]+num[j]*(n-1)>m) break;
            bool okay=1;
        
		    for(int k=1;k<=n-2;k++)
                if(!vis[num[j]+(num[j]-num[i])*k])
                {  
                    okay=0; 
                    break; 
                }
            
        if(okay) 
        {
            rl++;
            a[rl]=num[i];
            b[rl]=num[j]-num[i];
        }
    }

    if (rl==0)
	{
        cout<<"NONE"<<endl;
        return 0;
	}

    for(int i=1;i<=rl;i++)//应该先按b排序再按a排序
        for(int j=i+1;j<=rl;j++)
            if ((b[i]>b[j]) || ( (b[i]==b[j]) && (a[i]>a[j])))
            {
                swap(a[i],a[j]); 
			    swap(b[i],b[j]);
            }

    for(int i=1;i<=rl;i++)
        cout<<a[i]<<" "<<b[i]<<endl;
 
    return 0;
}