hdu 5389 Zero Escape(dp类似背包)

来源:互联网 发布:盐城办公软件培训 编辑:程序博客网 时间:2024/05/29 18:49

题目链接:

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;}



0 0