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


發佈了81 篇原創文章 · 獲贊 71 · 訪問量 8萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章