hdu 5389 Zero Escape(dp类似背包)

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=5389

解题思路：

题目大意：

有A和B两个门，每个门上有一个数字，现在有n个人，每个人手里也有一个数字，现在n个人都要选择一个门进去，要求计算门内的

人的数字每一位的和，一直累加到一位。这个数和门代表的数字是相同的。问n个人有多少种走法。

算法思想：

数字根直接mod 9就行了，随然公式是(x - 1) % 9 + 1，但是这里只要对不同的作区分即可，先判断(sum % 9)是不是不等于

(A + B) % 9，如果是，则显然只可能把人全放到一个门里，这时候有三种情况，一个门都不行；只能进一个门；和两个门值相同，

两个门都可以进，特判完，可以进行dp，dp[i][j]表示前i个人中进A门，数字根为j的可能数，这里不用管B，因为一共就两门，不是进

A就是进B，dp[i][(j + a[i]) % 9] = dp[i][(j + a[i]) % 9] + dp[i - 1][j]，即放或者不放，最后答案为dp[n][A % 9]。。。

AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

int const MOD = 258280327;
int const maxn = 100005;
int a[maxn],dp[maxn][11];

int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        int n, A, B;
        memset(dp,0,sizeof(dp));
        dp[0][0] = 1;
        int sum = 0;
        scanf("%d %d %d", &n, &A ,&B);
        for(int i = 1; i <= n; i++){
            scanf("%d", &a[i]);
            sum += a[i];
        }
        if(sum % 9 != (A + B) % 9){
            if(sum % 9 != A % 9 && sum % 9 != B % 9)
                printf("0\n");
            else
                printf("%d\n", A == B ? 2 : 1);
        }
        else{
            for(int i = 1; i <= n; i++){
                for(int j = 0; j < 9; j++)
                    dp[i][j] = (dp[i][j] % MOD + dp[i - 1][j] % MOD) % MOD;
                for(int j = 0; j < 9; j++)
                    dp[i][(j + a[i]) % 9] = (dp[i][(j + a[i]) % 9] % MOD + dp[i - 1][j] % MOD) % MOD;
            }
            printf("%d\n", dp[n][A % 9]);
        }
    }
    return 0;
}