题意: 有一个式子(a0*x^(2^0)+1) * (a1 * x^(2^1)+1)*.......*(an-1 * x^(2^(n-1))+1),给定一个P,求x^P的系数。
思路:根据推导得出:
例如 x^3 3的二进制表示为11,则x^3的系数为a[0]*a[1]
x^4 4的二进制表示为100,则x^4的系数为a[2]
对于本题输入n个ai,在将P化为二进制的过程中,当二进制位数大于n时,ai=0,所以x^P的系数为0。
#include <iostream>
#include <string.h>
#include <stdio.h>
using namespace std;
int main()
{
int t,a[105],n,q,m;
long long int p;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
for(int i=0; i<n; i++)
scanf("%d",&a[i]);
scanf("%d",&q);
while(q--)
{
long long int sum=1;
int t=0;
m=-1;
scanf("%lld",&p);
while(p)
{
t=p%2;
p/=2;
m++;
if(m>=n)
{
sum=0;
break;
}
if(t==1)
sum=(sum*a[m])%2012;
}
printf("%lld\n",sum);
}
}
return 0;
}