4008: [HNOI2015]亚瑟王

4008: [HNOI2015]亚瑟王

链接

分析:

  根据期望的线性性,直接求出每张牌出现的概率,最后乘以攻击力就是答案。

  每张牌出现的概率只与它前面的牌有关,与后面的没有关系,于是按顺序考虑每张牌。

  $f[i][j]$表示到第i张牌,还剩j次出牌的机会(即轮数)的概率,那么有$f[0][r] = 1.0$,然后考虑如何转移。

  $f[i][j] = f[i - 1][j] \times (1-p[i])^j + f[i - 1][j + 1] \times (1 - (1 - p[i])^{j+1})$

  第一项表示第i张牌没有造成伤害的概率,那么后面的轮数中,选到i的时候,概率都是$1-p[i]$,共j轮,所以是$(1-p[i])^j$。

  第二项表示第i张牌造成伤害的概率,那么后面的轮中,只要有一轮抽中即可,一共j+1轮,j+1轮都没有的概率是$(1-p[i])^{j +1}$,那么只要抽中一轮即可,就是$1-(1-p[i])^{j +1}$。

  第i张牌出现的概率就可以用第二项求出。

代码:

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iostream>
#include<cctype>
#include<set>
#include<map>
#include<queue>
#include<vector>
#define fi(s) freopen(s,"r",stdin)
#define fo(s) freopen(s,"w",stdout)
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 = 300;
double f[N][N], p[N], mi[N][N];
int a[N];
 
void solve() {
    int n = read(), r = read();
    for (int i = 1; i <= n; ++i) scanf("%lf%d", &p[i], &a[i]);
    for (int i = 1; i <= n; ++i) {
        mi[i][0] = 1.0;
        for (int j = 1; j <= r; ++j) mi[i][j] = mi[i][j - 1] * (1 - p[i]);
    }
    double ans = 0;
    memset(f, 0, sizeof(f));
    f[0][r] = 1.0;
    for (int i = 1; i <= n; ++i) {
        double now = 0;
        for (int j = r; ~j; --j) {
            f[i][j] = f[i - 1][j] * mi[i][j] + f[i - 1][j + 1] * (1 - mi[i][j + 1]);
            now += f[i - 1][j + 1] * (1 - mi[i][j + 1]);
        }
        ans += 1.0 * now * a[i];
    }
    printf("%.10lf\n", ans);
}
int main () {
    for (int T = read(); T --; solve());
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/mjtcn/p/10355618.html