「BZOJ 1426」收集邮票

题目链接

戳我

(Solution)

我们首先转换一下问题：

假设我们进行了k轮得到了所有种类的邮票

则所花费用为:

[(1+2+5+...+k)=frac{(1+k)*k}{2}=frac{k+k^2}{2} ]

所以我们现在要求的就是(frac{k+k^2}{2})的期望

因为(E(A+B)=E(A)+E(B))

所以(E(k+k^2)=E(k)+E(k^2))

(ps:)平方的期望不等于期望的平方

所以我们要分别维护

设(p1[i])为收集了(i)张邮票之后还要花费的次数的期望,(p2[i])为收集了(i)张邮票之后还要花费的次数平方的期望

由于有(frac{i}{n})的概率拿到重复的,(frac{n-i}{n})的概率拿到不重复的。

可得(p1)的方程：

[p1[i]=(p1[i]+1)*frac{i}{n}+(p1[i+1]+1)*frac{n-i}{n} ]

将上方程可化简为:

[p1[i]=p1[i+1]*frac{n}{n-i} ]

因为(E((x+1)^2)=E(x^2+2x+1)=E(x^2)+E(2x)+E(1)=E(X^2)+2E(x)+1)

(p2)的方程：

[p2[i]=(p2[i]+2*p1[i]+1)*frac{i}{n}+(p2[i+1]+p1[i+1]*2+1)*frac{n-i}{n} ]

化简为:

[p2[i]=p2[i+1]+2*p1[i+1]+1+(2*p1[i]+1)*frac{i}{n-i} ]

最后答案为:

[frac{p1[0]+p2[0]}{2} ]

(Code)

#include<bits/stdc++.h>
#define rg register
#define file(x) freopen(x".in","r",stdin);freopen(x".out","w",stdout);
using namespace std;
int read(){
    int x=0,f=1;char c=getchar();
    while(c<'0'||c>'9') f=(c=='-')?-1:1,c=getchar();
    while(c>='0'&&c<='9') x=x*10+c-48,c=getchar();
    return f*x;
}
double p1[10001],p2[10001];
int main(){
    int n=read();
    p2[n]=0;p1[n]=0;
    for(int i=n-1;i>=0;i--)
        p1[i]=p1[i+1]+n*1.0/(n-i),p2[i]=p2[i+1]+2*p1[i+1]+1+i*1.0/(n-i)*(2*p1[i]+1);
    printf("%0.2lf",(p1[0]+p2[0])/2);
    return 0;
}