bzoj 3751: [NOIP2014]解方程 同余系枚举

时间:2023-03-09 08:52:12
bzoj 3751: [NOIP2014]解方程 同余系枚举

3.解方程
(equation.cpp/c/pas)
【问题描述】
已知多项式方程:
a ! + a ! x + a ! x ! + ⋯ + a ! x ! = 0
求这个方程在[1, m]内的整数解(n 和 m 均为正整数)。
【输入】
输入文件名为 equation.in。
输入共 n+2 行。
第一行包含 2 个整数 n、m,每两个整数之间用一个空格隔开。
接下来的 n+1 行每行包含一个整数,依次为a ! , a ! , a ! , ... , a ! 。
【输出】
输出文件名为 equation.out。
第一行输出方程在[1, m]内的整数解的个数。
接下来每行一个整数,按照从小到大的顺序依次输出方程在[1, m]内的一个整数解。
【输入输出样例 1】
equation.in equation.out
2 10 1
1 1
-2
1
【输入输出样例 2】
equation.in equation.out
2 10 2
2 1
-3 2
1
【输入输出样例 3】
equation.in equation.out
2 10 0
1
3
2
第 5 页共 6 页提高组 day2
全国信息学奥林匹克联赛(NOIP2014)复赛
【数据说明】
对于 30%的数据,0 < n ≤ 2, a ! ≤ 100,a ! ≠ 0, m ≤ 100;
对于 50%的数据,0 < n ≤ 100, a ! ≤ 10 !"" ,a ! ≠ 0,m ≤ 100;
对于 70%的数据,0 < n ≤ 100, a ! ≤ 10 !"""" ,a ! ≠ 0,m ≤ 10000;
对于 100%的数据,0 < n ≤ 100, a ! ≤ 10 !"""" ,a ! ≠ 0,m ≤ 1000000。

  考场上想到了在同余系中枚举x,但是对于最后30%数据用的自然溢出,自然溢出很容易被卡掉。其实这种做法已经十分贴近正解了。观察发现,取质数p,对于解a,b,若a==b(mod p)那么他们带入算出的值一定相同,所以只用取多个较小p,预处理出[0,p-1]的值,在枚举[1,m]代入检验即可,bzoj数据很不良心啊,要么TLE,要么WA,jason_yu帮我检查了半天,才发现由于ans=(ans * i + a[ii][j])%pmod;我写成了{ans=(ans+x*a[j])%pmod;x=x*i%pmod;}常数过大,所以时限内尝试质数个数过少。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
using namespace std;
#define PROB "equation"
#define MAXN 1100001
#ifdef WIN32
#define LL "%I64d"
#else
#define LL "%lld"
#endif
typedef long long qword;
const int mod[] =
{,,,,
};
int a[][];
bool ok[MAXN];
bool ok2[][];
char s[];
int n,m;
void work();
int main()
{
freopen(PROB".in","r",stdin);
freopen(PROB".out","w",stdout);
int ii,i,j, k;
int x;
scanf("%d%d\n",&n,&m);
int ll = ;
for (ii=;ii<=n;ii++)
{
// scanf("%s", s);
fgets(s,sizeof(s),stdin);
int l = strlen(s)-;
if((s[l-]<'' || s[l-]>'') && s[l-]!='-')l--;
for(int j = ; j < ll; j ++){
int pmod = mod[j];
a[j][ii] =;
x=;
k=;
if (s[k]=='-')k++,x=-x;
else if (s[k]=='+')k++;
for (;k<l;k++)
{
a[j][ii]=(a[j][ii]*+s[k]-'')%pmod; }
a[j][ii]=a[j][ii]*x%pmod;
if(a[j][ii] < )
a[j][ii] += pmod;
}
}
memset(ok,,sizeof(ok));
memset(ok2,,sizeof(ok2));
int pmod;
int ans;
for (ii=;ii<ll;ii++)
{
pmod=mod[ii];
for (i=;i<pmod;i++)
{
ans=;
for (j=n;j>=;j--)
{
ans=(ans * i + a[ii][j])%pmod;
}
if (ans)
{
ok2[ii][i]=false;
}
}
}
int cnt = ;
for (int i=;i<=m && cnt <= n;i++)
{
for (int j=;j<ll;j++)
{
if (!ok2[j][i%mod[j]]){
ok[i]=false;
break;
}
}
if(ok[i])
++ cnt;
}
printf("%d\n",cnt);
for (i=;i<=m;i++)
if (ok[i])
printf("%d\n",i);
return ;
}