hdu 5464 Clarke and problem(DP)

題目大意：給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;
}