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



原文地址:https://www.cnblogs.com/llguanli/p/7130621.html