hdu5389

题意:给你n个人每个人手里有一个id,然后给你两个数a和b,让你把n个人分为两组,条件是 一组人手里的id等于a 另一组人的id等于b,这里的是指加起来之后对9取余,如果sum等于0 则sum等于9 否则sum = sum;还有一种情况也可以 就是所有人的id等于a 或者等于b 相当于分为一组。

思路:首先 如果能找到满足题意的解,一定满足a和b的等于n个人的标号的

          然后  数位dp,dp[i][j]表示在第i个数字的时候 前面i个数字能组成j并且包括arr[i]的方案数。

          状态转移方程:dp[i][j] = dp[i-1][j] + dp[i-1][cnt],cnt = j - arr[i](这个应该能理解吧),cnt<=0时要加上9;

          另外 如果a和b的不等于n个人的标号的还要考虑是否满足 a == sum或b == sum;

代码:

#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <cstdio>
#include <string>
#include <bitset>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <list>
#include <map>
#include <set>
#define sss(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a) memset(a,0,sizeof(a))
#define ss(a,b) scanf("%d%d",&a,&b)
#define s(a) scanf("%d",&a)
#define p(a) printf("%d\n", a)
#define INF 0x3f3f3f3f
#define w(a) while(a)
#define PI acos(-1.0)
#define LL long long
#define eps 10E-9
#define N 1000000+20
#define mod 258280327
const int SIGMA_SIZE=26;
const int MAXN=100010;
const int MAXNODE=600010;
using namespace std;
void mys(int& res)
{
    int flag=0;
    char ch;
    while(!(((ch=getchar())>='0'&&ch<='9')||ch=='-'))
        if(ch==EOF)  res=INF;
    if(ch=='-')  flag=1;
    else if(ch>='0'&&ch<='9')  res=ch-'0';
    while((ch=getchar())>='0'&&ch<='9')  res=res*10+ch-'0';
    res=flag?-res:res;
}
void myp(int a)
{
    if(a>9)
        myp(a/10);
    putchar(a%10+'0');
}
/*************************THE END OF TEMPLATE************************/
int arr[100009];
int dp[100009][10];
int sum_mod(int x, int y){
    int ans = x + y;
    ans %= 9;
    if(!ans) return 9;
    return ans;
}
int main(){
    int t, n, a, b;
    s(t);
    w(t--){
        mem(dp);
        sss(n, a, b);
        int sum = 0;
        for(int i=1; i<=n; i++){
            s(arr[i]);
            sum = sum_mod(sum, arr[i]);
        }
        if(sum != sum_mod(a, b)){
                int ans = 0;
                if(a == sum) ans ++;
                if(b == sum) ans ++;
                p(ans % mod);
        }
        else{
            dp[1][arr[1]] = 1;
            for(int i=2; i<=n; i++){
                for(int j=1; j<=9; j++){
                    int cnt = j - arr[i] <= 0? j - arr[i] + 9: j-arr[i];
                    dp[i][j] = (dp[i-1][j] + dp[i-1][cnt]) % mod;
                }
            }
            p((dp[n][a]+dp[n][b])%mod);
        }
    }
    return 0;
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值