F - Pass

鏈接

題意：

　　給定一個只包含012序列，0表示這個人有2個紅球，1表示一個紅球一個藍球，2表示兩個藍球。進行n*2次遊戲，每次遊戲所有有球的人選擇一個球遞給前一個人，第一個人把球放到一個序列中，‘r‘表示紅球，‘b‘表示藍球，問一共多少種序列。

分析：

　　一定要滿足前i個人的所有紅球>=序列前i個位置的紅球，藍球同理，然後dp。

　　f[i][j]表示到第i個位置，有j個紅球的方案數。

代碼：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include
 
<iostream>
#include<cmath>
#include<cctype>
#include<set>
#include<queue>
#include<vector>
#include<map>
using namespace std;
typedef long long LL;

inline int read() {
    int x=0,f=1;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch==‘-‘)f=-1
 
;
    for(;isdigit(ch);ch=getchar())x=x*10+ch-‘0‘;return x*f;
}

const int N = 4005, mod = 998244353;
int sum[N][2], f[N][N];
char s[N];

inline void add(int &x,int y) {
    x += y; x = x >= mod ? x - mod : x;
}

int main() {
    scanf("%s", s + 1);
    int n = strlen(s + 1);
    for
 
 (int i = 1; i <= n; ++i) {
        if (s[i] == ‘0‘) sum[i][0] += 2;
        else if (s[i] == ‘1‘) sum[i][0] ++, sum[i][1] ++;
        else sum[i][1] += 2;
        sum[i][0] += sum[i - 1][0], sum[i][1] += sum[i - 1][1];
    }

    if (sum[1][0] >= 1) f[1][1] = 1;
    if (sum[1][1] >= 1) f[1][0] = 1;
    
    for (int i = 2; i <= (n << 1); ++i) {
        for (int j = 0; j <= i; ++j) {
            int r = j, b = i - j;
            if (r > sum[min(i, n)][0] || b > sum[min(i, n)][1]) continue;;
            add(f[i][j], f[i - 1][j]);
            if (j) add(f[i][j], f[i - 1][j - 1]);
        }
    }
    cout << f[n + n][sum[n][0]];
    return 0;
}

Yahoo Programming Contest 2019 F - Pass