HDU 5389 Zero Escape

来源：互联网发布：mac jdk1.7 下载编辑：程序博客网时间：2024/06/04 23:54

题意：给出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 258280327int 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;}

0 0