题目大意:给n个数,问选择其中若干个数使得和能够被p整除的方案数。
设dp[i][j]表示前i个数选取若干个,和被p除余j的方案数。
那么dp[i][j]=dp[i-1][j]+dp[i-1][j-a[i]%p] (j=0~p-1)
坑点:a[i]可以为负数,取模的时候注意一下。
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<map>
#include<set>
#include<queue>
#include<stack>
using namespace std;
typedef double db;
typedef __int64 LL;
#define mod 1000000007
LL a[1005],dp[1001][1000];
inline int mo(int a,int p){
if(a>=0) return a%p;
return a%p+p;
}
int main()
{
int T,n,p,i,j;
cin>>T;
while(T--)
{
scanf("%d%d",&n,&p);
memset(dp,0,sizeof(dp));
dp[0][0]=1;
for(i=1;i<=n;++i)
{
scanf("%I64d",&a[i]);
a[i]=mo(a[i],p);
for(j=0;j<p;++j) dp[i][j]=(dp[i-1][j]+dp[i-1][(j-a[i]+p)%p])%mod;
}
printf("%I64d\n",dp[n][0]);
}
return 0;
}