LightOJ 1274 Beating the Dataset (概率dp)

题意：做n个题目蒙答案，答案要么是YES要么是NO，给出分别有多少个。第一题猜YES，然后接下来每一题猜上一题的正确答案，求不通过的数据点期望数量。

题解：概率dp
可以解方程求出YES和NO的数量a和b。
$dp[i][j][0/1]$表示当前第$i$题，已经输出了$j$个YES（包括当前），当前要输出YES/NO。
因为第一个猜YES，那么相当于正确答案前加一个YES，比较答案的相邻值，结果就是$dp[0][0][0]$。
若当前要输出YES，i+1行如果输出NO/YES：
dp[i][j][0] = (dp[i + 1][j][1] + 1) * pn + dp[i + 1][j + 1][0] * py
若当前要输出NO，i+1行如果输出YES/NO：
dp[i][j][1] = (dp[i + 1][j + 1][0] + 1) * py + dp[i + 1][j][1] * pn
当然这样数组会超，用滚动数组就好了。

这题还可以推公式，不写了。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
int t, n, s, a, b;
double dp[2][5555][2];
int main() {
	scanf("%d", &t);
	int cas = 0;
	while (t--) {
		scanf("%d%d", &n, &s);
		a = s - 2 * n, b = n - a;
		memset(dp, 0, sizeof(dp));
		for (int i = n - 1; i >= 0; i--) {
			int minn = max(i - b, 0), maxn = min(i, a);
			//memset(dp[i % 2], 0, sizeof(dp[i % 2]));
			for (int j = minn; j <= maxn; j++) {
				double py = 1.0 * (a - j) / (n - i);
				double pn = 1.0 * (b - (i - j)) / (n - i);
				dp[i % 2][j][0] = (dp[(i + 1) % 2][j][1] + 1) * pn + dp[(i + 1) % 2][j + 1][0] * py;
				dp[i % 2][j][1] = (dp[(i + 1) % 2][j + 1][0] + 1) * py + dp[(i + 1) % 2][j][1] * pn;
			}
		}
		printf("Case %d: %f\n", ++cas, dp[0][0][0]);
	}
	return 0;
}