题目描述
学了数论后,小杰最近对整系数多项式在整点的取值感兴趣,他现在来求助你帮他计算。
一个整系数多项式通常表示为f(x)=a0+a1x+a2x2+a3x3+⋯+anxn,其中ai(i=0,1,2,⋯,n)为整数。
给定x∈Z,求出f(x)mod1000000007。输入描述:
多组读入。
每组的第一行,表示多项式f(x)
接下来是一个整数n,接着一行有n个整数x,你需要求值f(x) mod 1000000007
组数不超过100,x在int表示的范围内,f(x)所有项的系数和指数都在int表示的范围内(指数保证非负)。输出描述:
对于每一个,在单独的一行中输出一个整数。
示例1
输入3x^2+6x+10-3x
4
0
1
2
3输出
10
16
28
46备注:
65%左右的多项式项数不超过20,查询数不超过20
25%左右的多项式项数不超过100,查询数不超过25
10%左右的多项式项数不超过10000,查询数不超过100
注意:对于f(x)的单项式aix^i(i!=0),若系数ai为0,我们有时会忽略这些项,若i=1,可能表示为a1x^i或是a1x。因为某些原因,对于幂次相同的单项式可能出现多次。
#include<bits\stdc++.h>
using namespace std;
#define ll long long
const ll mod = 1e9 + 7;
const ll MAXN = 1e8;
char s[MAXN];
int A[MAXN] , B[MAXN] , n = 0 , q;
void ini ()
{
n = 0;
int t = 0 , x[2] = { 0,0 } , y[2] = { 0,0 } , f = 1;
int len = strlen (s);
s[len++] = '+';
for (int i = 0; i<len; i++)
{
if (s[i] <= '9'&&s[i] >= '0')
{
x[t] = x[t] * 10 + s[i] - '0';
y[t] = 1;
}
else if (s[i] == 'x')
t = 1;
else if (s[i] == '+' || s[i] == '-')
{
if (t == 1 && !y[1])
x[1] = 1;
if (t == 1 && !y[0])
x[0] = 1;
A[n] = (x[0] * f % mod + mod) % mod;
B[n] = x[1];
n++;
y[0] = y[1] = x[0] = x[1] = t = 0;
f = s[i] == '+' ? 1 : -1;
}
}
}
inline int powf (int a , int b)
{
ll ans = 1;
for (; b; b >>= 1 , a = (ll)a*a%mod)
if (b & 1) ans = ans * a % mod;
return ans;
}
int main ()
{
while (scanf ("%s" , s) != EOF)
{
ini ();
for (scanf ("%d" , &q); q; q--)
{
ll ans = 0 , t;
scanf ("%lld" , &t);
t = (t % mod + mod) % mod;
for (int i = 0; i<n; i++)
ans = (ans + A[i] * powf (t , B[i])) % mod;
printf ("%lld\n" , ans);
}
}
}
我们用A[]保存系数,B[]保存指数然后一个个项按个运算过去即可。