USACO Section 1.4 ariprog - 需要加点点优化搜索题

    题目的意思开始看了好久~~~其实看懂了还好写....先将bisquares数打个表出来...这里为了方便搜索..我是打了两个表...一个是打出1~200000中每个数是不是bisquares，是的话组成这个数的最大p会是多少...另一个表是依次存着在其最大p值不大于M的bisquares.....

    搜索的时候先遍历打出的bisquares表枚举队首...

    然后再遍历bisquares表枚举第二个数

    这时就得出差了...然后就一步步走到a+(n-1)b...每走一步就用大户的1~200000那个表来判断下是否满足条件...

    输出的时候先给sort一下~~就行了...

Program:

/*  
ID: zzyzzy12  
LANG: C++   
TASK: ariprog 
*/    
#include<iostream>    
#include<stdio.h>    
#include<string.h>    
#include<math.h>    
#include<algorithm>    
#include<queue>
using namespace std;
struct node
{
     int a,b;  
}ans[10001];
int f[150000];
int i,j,N,M,k,y,p,s[150001],num;
bool cmp(node a,node b)
{
     if (a.b==b.b) return a.a<b.a;
     return a.b<b.b;    
}
int main()
{
     freopen("ariprog.in","r",stdin);
     freopen("ariprog.out","w",stdout);
     scanf("%d%d",&N,&M);  
     memset(f,-1,sizeof(f)); 
     for (i=0;i<=M;i++)
       for (j=i;j<=M;j++) 
          f[i*i+j*j]=j;
     k=0;     
     for (i=0;i<=150000;i++)
     if (f[i]>=0) 
     {
          k++;      
          s[k]=i;    
     }
     num=0;
     for (i=1;i<=k;i++) 
     {  
          for (j=i+1;j<=k;j++)  
          {
                y=s[j]-s[i]; 
                for (p=2;p<N;p++)
                   if (s[i]+y*p>2*M*M || f[s[i]+y*p]<0 || f[s[i]+y*p]>M) goto A; 
                num++;
                ans[num].a=s[i]; ans[num].b=y;
                A: ;
          } 
     }
     if (num)
     {
          sort(ans+1,ans+1+num,cmp);
          for (i=1;i<=num;i++) printf("%d %d\n",ans[i].a,ans[i].b);
     }else printf("NONE\n");
     return 0;   
}