洛谷P1214 [USACO1.4]等差数列 Arithmetic Progressions

题目描述 https://www.luogu.org/problemnew/show/P1214
如果直接枚举a,b会超时，可以枚举数列的前两项，将双重平方数【注意：小心重复】从大到小排序，枚举前两项，可得到公差b,在依次-b，判断这个数是否是双重平方数，最后按题目要求排序。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
int a[200010],tot,num,n,m;
bool fl[200010];
struct node
{
    int a,b;
}ans[200010];
bool cmp(int x,int y)
{
    return x>y;
}
bool tmp(node x,node y)
{
    if(x.a!=y.a) return x.a<y.a;
    return x.b<y.b;
}
int main( )
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<=m;i++)
      for(int j=i;j<=m;j++)
	    if(!fl[i*i+j*j]) a[++tot]=i*i+j*j,fl[i*i+j*j]=1;
    sort(a+1,a+tot+1,cmp);
    for(int i=1;i<=tot-n+1;i++)
      for(int j=i+1;j<=tot-n+2;j++)
	  {           
         int d=a[i]-a[j],cnt=n-2,t=a[j],fg=1;
         if(t-(n-2)*d<0)break;        
         while(cnt)
		 {
            cnt--,t-=d;
            if(t<0){ fg=0; break;}
            if(!fl[t]){ fg=0; break;}
         }
         if(fg) ans[++num].a=d,ans[num].b=t;
      }
    if(!num){ printf("NONE\n"); return 0;}
    sort(ans+1,ans+num+1,tmp);
    for(int i=1;i<=num;i++) printf("%d %d\n",ans[i].b,ans[i].a);
    
 return 0;
}