NOIP2014 解方程

题目描述

已知多项式方程：

a0+a1x+a2x^2+..+anx^n=0

求这个方程在[1, m ] 内的整数解（n 和m 均为正整数）

输入输出格式

输入格式：
输入文件名为equation .in。
输入共n + 2 行。
第一行包含2 个整数n 、m ，每两个整数之间用一个空格隔开。
接下来的n+1 行每行包含一个整数，依次为a0,a1,a2..an
输出格式：
输出文件名为equation .out 。
第一行输出方程在[1, m ] 内的整数解的个数。
接下来每行一个整数，按照从小到大的顺序依次输出方程在[1, m ] 内的一个整数解。

输入输出样例

输入样例#1：
2 10
1
-2
1
输出样例#1：
1
1
输入样例#2：
2 10
2
-3
1
输出样例#2：
2
1
2
输入样例#3：
2 10
1
3
2

输出样例#3：
0

代码

#include
#include
#define M 1000001
#include
using namespace std;
int p[6]={0,7457,7829,8623,8971,9781};
int pre[6][105],a[6][105],ans[M],res[6][30005];
char c[10000];
int n,m;
inline int read() {
    int x=0,f=1;char c=getchar();
    while(c>'9'||c<'0') {if(c=='-')f=-1;c=getchar();} 
    while(c>='0'&&c<='9') x=10*x+c-48,c=getchar();
    return x*f;
}
inline int cal(int t) {
    int sum=0;
    for(int i=0;i<=n;i++) {
        sum+=(a[t][i]*pre[t][i])%p[t];
        sum%=p[t];
    }
    if(sum<0) sum+=p[t];
    return sum;
}
inline bool judge(int x) {
    for(int i=1;i<=5;i++)
      if(res[i][x%p[i]]) return false;
    return true;
}
int main() {
    bool flag=false;
    n=read();m=read();
    for(int i=0;i<=n;i++) {
        flag=false;
        scanf("%s",c+1);
        int l=strlen(c+1);
        for(int t=1;t<=5;t++)
          if(c[1]!='-') a[t][i]=c[1]-48;
            else a[t][i]=0,flag=true;
        for(int t=1;t<=5;t++) {
            for(int k=2;k<=l;k++)
                a[t][i]=(a[t][i]*10+c[k]-48)%p[t];
            if(flag) a[t][i]=-a[t][i];
        }
    }
    for(int j=1;j<=5;j++) 
      for(int k=1;k0]=1;
        for(int i=1;i<=n;i++)
          pre[j][i]=(pre[j][i-1]*k)%p[j];
        res[j][k]=cal(j);
      }
    for(int i=1;i<=m;i++)
      if(judge(i)) ans[++ans[0]]=i;
    printf("%d\n",ans[0]);
    for(int i=1;i<=ans[0];i++)
      printf("%d\n",ans[i]);
    return 0;
}