I - Beating the Dataset
LightOJ - 1274题意:一个答案文件有n行输出,每行是YES或NO,现在给出文件的总字节数s,YES占3字节,NO占2字节。
要求你输出1个文件,且第一行一定是NO,接下来第i行的内容是答案文件第i-1行的内容,问输出正确行数的期望
题解:期望dp,用逆推。dp[i][j][k]表示当前状态为第i行,已经输出了j个YES,这一行要输出YES(0)/NO(1)
因此可以得出,①如果第i要输出YES,那么i+1行如果输出NO,则i的期望值为i+1的期望值加上+1再乘以输出NO的概率;i+1行如果输出YES,则i的期望值为i+1的期望值乘以输出YES的概率。②如果第i行要输出NO,做法和①相同
递推关系式为dp[i][j][0]=dp[i+1][j+1][0]*p1+(dp[i+1][j][1]+1)*p2
dp[i][j][1]=(dp[i+1][j+1][0]+1*p1)*p1+dp[i+1][j][1]*p2
因为不能开5000*5000*2的数组,想到第i位只与第i+1位有关,第一维可以滚动。
#include<cstdio>
#include<string.h>
#include<algorithm>
using namespace std;
const int MX = 5e3 + 5;
double dp[2][MX][2];
int main() {int n, s, T;// freopen ("in.txt", "r", stdin);scanf ("%d", &T);for (int cas = 1; cas <= T; cas++) {scanf ("%d%d", &n, &s);int yes = s - 2 * n;int no = n - yes;memset (dp, 0, sizeof (dp));double ans = 0;for (int i = n - 1; i >= 0; i--) {int ret = n - i, now = i % 2, nxt = (i + 1) % 2;int Max = min (i, yes), Min = max (0, i - no); //前i个文件中yes的个数为[Min,Max]memset (dp[now], 0, sizeof (dp[now]));for (int j = Min; j <= Max; j++) {double p1 = 1.0 * (yes - j) / ret, p2 = 1.0 * (no - (i - j)) / ret;dp[now][j][0] = dp[nxt][j + 1][0] * p1 + (dp[nxt][j][1] + 1) * p2;dp[now][j][1] = (dp[nxt][j + 1][0] + 1) * p1 + dp[nxt][j][1] * p2;}}printf ("Case %d: %.7f\n", cas, dp[0][0][0]);//第一行只能输出YES}return 0;
}