HDU 5389 Zero Escape

题意：给出A, B两个值， 再给出n个数，问n个数分成2组，使其中一组的数字根为A，另一组数字根为B，或者所有数的数字根为A或B的方法有多少种。</span>

 数字根题目有说，还可以直接把数加起来然后模9得到。</span>

思路：dp[i][j]表示前i个数中，组成的数字根值为j的方法有多少种</span>

dp[i][j] = (dp[i][j] + dp[i - 1][j]) % mod;  

dp[i][fun(j, a[i])] = (dp[i][fun(j, a[i])] + dp[i - 1][j]) % mod;  

更新添加上a[i]之后能得到的数字根的方法数量 </span>

最后要处理一下：能得到A, B，或者所有的数加起来得到的数字根只能得到A或B的情况

#include <stdio.h>
#include <string.h>
#define maxn 100010
#define mod 258280327
int dp[maxn][10], a[maxn];
int fun(int a, int b)
{
    int res = a + b;
    res %= 9;
    if(res == 0) return 9;
    return res;
}
int main()
{
    int t, n, sum, A, B, i, j, ans, tp;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d %d %d", &n, &A, &B);
        sum = 0;
        for(i = 1;i <= n;i++)
        {
            scanf("%d", &a[i]);
            sum = fun(sum, a[i]);
        }
        memset(dp, 0, sizeof(dp));
        for(i = 1;i <= n;i++) dp[i][a[i]] = 1;

        for(i = 1;i <= n;i++)
        {
            for(j = 0;j < 10;j++)
            {
                dp[i][j] = (dp[i][j] + dp[i - 1][j]) % mod;
                dp[i][fun(j, a[i])] = (dp[i][fun(j, a[i])] + dp[i - 1][j]) % mod;
            }
        }
        tp = fun(A, B);
        ans = 0;
        //printf("%d\n", sum);
        if(sum == tp)
        {
            ans += dp[n][A];
            if(tp == A) ans--;
        }
        if(sum == A) ans++;
        if(sum == B) ans++;
        printf("%d\n", ans);
    }
    return 0;
}