洛谷P2111 考場的奇遇(概率dp or 記憶化搜索)

2020.6.22
今天心情不好，因爲製造了一堆學術垃圾。今天一天的晦氣是從早上不自覺多睡了一個小時開始的，喫完飯下午開始來寫東西，結果一連寫了三道題都是70分，這可氣死我了。都是rand來的題單。

這道題思路非常赤裸，就是簡簡單單的二項式分佈，icpc考的很多，比如去年的researchability index。 然而好像並不用優化來着，直接遞歸走人（然而那時候我只寫了數據結構，沒打出來，事後看看不應該丟給隊友的，有兩道題都是事後發現很簡單，但沒輪到我那種）。

這個問題二項式分佈就是n道題做對k道的概率，有個約束條件就是兩個人有相同的和不同的題，所以這個比較拐彎，但還不至於太難想，比較可惡的是出題人說這是數論和搜索。我承認這是記憶化，但絕非數論內容啊。首先就是求期望。分解每一步要考慮的是n道題做對了幾道，兩人相同的情況下，首先如果在n道題中做對了k道題有兩種情況，即上一道題（n - 1）沒有做對，此時對的題目爲k - 1，這道做對了，我們乘上p的概率，如果n-1做對了這道沒對，那就乘上np。二者相加即爲n和k的概率。

反之，如果不同，n - 1沒做對，那就得乘上np， 如果做對了，那麼就乘上p，二者相加即爲我們需要的答案。注意最後特例要特判，因爲數組開不下。。。。

我感覺這思路沒問題，怎麼就wa了啊。

#include <bits/stdc++.h>
using namespace std;
#define limit (10000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步兩步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%d\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 998244353
#define FOPEN freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\data.txt", "rt", stdin)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快讀
void write(ll x){
    if(x < 0) putchar('-'),x = -x;
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}

int n,m,k,tot;
int a[limit];
double dp[limit][50 + 5];//做i道對j道的
int main() {
#ifdef LOCAL
    FOPEN;
#endif
    n = read(),m = read(), k = read();
    double p = 1.0 * m / 100;//概率
    double np = 1 - p;
    rep(i ,1,n){
        char c;
        scanf("%c" , &c);
        a[i] = c - '0';
    }
    if(!k)return 0 * puts("1.000");
    dp[0][0] = 1.00;
    rep(i ,1, n){//從上一題過來然後寫錯
        dp[i][0] = dp[i - 1][0] * (a[i] ? np : p);//
    }
    rep(i ,1, n){
        rep(j ,1, i){
            if(!a[i])dp[i][j] = dp[i - 1][j] * p + dp[i - 1][j - 1] * np;//如果不一樣就是要麼上一題的時候做錯了，
            else dp[i][j] = dp[i - 1][j] * np + dp[i - 1][j - 1] * p;
        }
    }
//    rep(i ,1, n){
//        rep(j ,1, i){
//            printf("at %d and %d we have%lf    ",i,j, dp[i][j]);
//        }
//        cout<<endl;
//    }
    double ans = 0.0000;
    per(i , k , n){
        ans += dp[n][i];
    }
    printf("%.3lf", ans);
    return 0;
}