Yahoo Programming Contest 2019 F - Pass
阿新 • • 發佈:2019-02-14
amp else string 鏈接 main 一個人 scan cout cto
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