LightOJ 1125 Divisible Group Sums （dp）

Description
求从n个整数a[i]中选出m个数使得这m个数之和被d整除的方案数
Input
第一行一整数T表示用例组数，每组用例输入三个整数n,q分别表示序总元素个数和查询数，之后输入n个整数a[i]表示这n个元素，每次查询给出两个整数d和m表示查询选m个数使得其和被d整除的方案数
(T<=20,1<=n<=200,1<=q<=10,1<=d<=20,1<=m<=10)
Output
对于每次查询输出结果
Sample Input
2
10 2
1
2
3
4
5
6
7
8
9
10
5 1
5 2
5 1
2
3
4
5
6
6 2
Sample Output
Case 1:
2
9
Case 2:
1
Solution
dp[i][j][k][l]表示从前i个数中选j个数使得其和模k为l的方案数，那么根据是否选取第i个数有
dp[i+1][j][k][l]+=dp[i][j][k][l]
dp[i+1][j+1][k][(l+a[i+1]%k+k)%k]+=dp[i][j][k][l];
对于每次查询，结果即为dp[n][m][d][0]，时间复杂度O(Tnmd^2)
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
ll a[222],dp[222][22][22][22];
int T,n,q,d,m,res=1;
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
        memset(dp,0,sizeof(dp));
        for(int k=1;k<=20;k++)dp[0][0][k][0]=1;
        for(int i=0;i<n;i++)
            for(int j=0;j<=min(i,10);j++)
                for(int k=1;k<=20;k++)
                    for(int l=0;l<k;l++)
                    {
                        dp[i+1][j][k][l]+=dp[i][j][k][l];
                        dp[i+1][j+1][k][(l+a[i+1]%k+k)%k]+=dp[i][j][k][l];
                    }
        printf("Case %d:\n",res++);
        while(q--)
        {
            scanf("%d%d",&d,&m);
            printf("%lld\n",dp[n][m][d][0]);
        }
    }
    return 0;
}