题意:给你n个数,将这n个数分成两部分,第一部分满足所有数字和的数字根为A,第二部分为B(数字根:将一个数的各个位加起来,如果和大于10,继续加,直到小于10)ps:一个数字的数字根等于各位数字和%上9;如果答案为0,则为9,其余则为模下来的值。
解题思路:本题关键是要知道数字根,然后很容易想到用dp来转移,我们用dp[i-1][j][k],表示前i-1个数字被分到两端,使得两端的状态为j,k,那么转移方程为dp[i][j][y]+=dp[i-1][j][k],dp[i][x][k]+=dp[i][j][k],(x=(j+a[i])%9,y=(k+a[i])%9,ps:注意x,y=0是的特判)可知复杂度为n*10*10,这样是可能超时的,仔细一想我们只需算出一边的情况即可,即:dp[i-1][j],表示前i-1项,左边状态为j的方法数,转移dp[i][j]+=dp[i-1][j],dp[i][x]=dp[i-1][j],这样我们就减少了10倍的复杂度
代码:
//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<iostream>
#include<map>
#include<set>
#include<stack>
#include<vector>
#include<queue>
#include<cmath>
#include<algorithm>
using namespace std;
#define inf 0x3fffffff
//const double pi=acos(-1);
typedef long long LL;
#define mod 258280327
int aa[100005];
int dp[100005][15];
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,a,b;
scanf("%d%d%d",&n,&a,&b);
int sum=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&aa[i]);
sum+=aa[i];
if(sum>9)
sum-=9;
}
memset(dp,0,sizeof(dp));
int t=0;
dp[0][0]=1;
for(int i=1;i<=n;i++)
{
for(int j=0;j<=9;j++)
{
int x=(j+aa[i])%9;
if(x==0)x=9;
dp[i][j]=(dp[i][j]+dp[i-1][j])%mod;
dp[i][x]=(dp[i][x]+dp[i-1][j])%mod;
}
}
int ans=0;
int xx=(a+b)%9;
if(xx==0)xx=9;
if(xx==sum)
{
ans+=dp[n][a];
if(ans>=mod)ans%=mod;
if(a==sum)
ans--;
}
if(a==sum)
{
ans++;
}
if(b==sum)
{
ans++;
}
if(ans>=mod)ans%=mod;
printf("%d\n",ans);
}
return 0;
}