HDU 5389 Zero Escape

HDU 5389 Zero Escape

/**
简单dp但是处理的细节多一些
状态就是用当前ai和上次的dp[i-1][j:0-9]
dp[i][index(j+a[i])]+=dp[i][j];有当下数字和之前装太得到的
dp[i][j]+=dp[i-1][j];//通过上次过继过来的
*/
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#define maxn 100005
#define ll __int64
#define INF 0xffffff
using namespace std;
ll dp[maxn][10];
ll c[maxn];
const ll mod=258280327;
ll index(ll num){ //求树根
    num%=9;
    return num==0?9:num;
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        int n,a,b;
        scanf("%d%d%d",&n,&a,&b);
        for(int i=1;i<=n;i++) scanf("%I64d",&c[i]);
        memset(dp,0,sizeof(dp));
        dp[0][0]=1;
        ll sum=0;
        for(int i=1;i<=n;i++){
            sum+=c[i];
            for(int j=0;j<10;j++) dp[i][j]=dp[i-1][j];
            for(int j=0;j<10;j++){
                dp[i][index(j+c[i])]+=dp[i-1][j];
                dp[i][index(j+c[i])]%=mod;
            }
        }
        ll ans=0;
        if(index(sum)==index(a+b)){
            ans+=dp[n][a];
            if(index(sum)==a)
                ans--;
        }
        if(a==index(sum)) ans++;
        if(b==index(sum)) ans++;
        printf("%I64d\n",ans);
    }
    return 0;
}