2022ICPC昆明 C题-Cup of Water 题解

geruome

已于 2022-07-17 01:44:06 修改

阅读量550

点赞数 2

文章标签：算法概率论

于 2022-07-17 01:40:42 首次发布

本文链接：https://blog.csdn.net/weixin_52867134/article/details/125827382

版权

题意： $T (T <= 10000)$ 组询问 , 每组给定 $\le x \le 1e9)$ . 每次从 $[0, x]$ 随机抽一个实数，累加这些实数，直到和 $\ge 1$ 停止抽。求抽取次数的期望。

思路：
step1 :
记抽取次数为 $k$ , $\ge i)-P(k \ge i+1)$ 。我们只需计算出所有 $\ge i)$ 即可。
进一步化简， $\sum\limits_{i=1}^\infty i*P(k=i) = \sum\limits_{i=1}^\infty P(k\ge i)$ 。
step2 :
$\ge i+1)$ = $P (抽取 i 次仍 < 1)$ = $P(\sum\limits_{j=1}^ix_j < 1),0<x_j<x$ 。
首先考虑简单点的情况 $x > 1$ 。
$i = 1$ : $P (抽取 i 次仍 < 1)$ = $1/ x$ ; 长度为1.
$i = 2$ : $P (抽取 i 次仍 < 1)$ = $1/(2*x^2)$ ; 三角形面积为1/2.
$i = 2$ : $P (抽取 i 次仍 < 1)$ = $1/(6*x^3)$ ; 三棱锥体积为1/6.
猜测出结论： $\sum\limits_{j=1}^ix_j < len$ 作为 $i$ 重积分区域时，测度为 $\frac{len^i}{i!}$ $0<x_j<x且x>len）$ 。
此式记为 $f (l e n)$ ;
step3:
当 $x < 1$ : 问题在于计算出 $\sum\limits_{j=1}^ix_j < 1（0<x_j<x）$ 的测度。
我们要利用step2中的测度公式，那么不妨先将每个抽取的 $x_j$ 的范围扩大到[0, $\infty$ ] 。
假设 $i$ 个数中有 $j$ 个数 $> x$ 的测度为 $ans_j$ 。我们要计算 $ans_0$ 。
可以用容斥来解决。
选 $j$ 个数 $> x$ ,其余数范围不限（即为[0, $\infty$ ] ）。把这 $j$ 个数的 $a$ 从总和中剔除掉，则选定一组 $j$ 个数后，测度为 $f (1 - ja)$ 。（前提是 $1 - ja > 0$ ）
则 $f(1-ja)*C(n,j)=C(j,j)*ans_j+C(j+1,j)*ans_{j+1}+C(j+2,j)*ans_{j+2}+.....$
$ans_0$ 可用容斥计算，结果为 $\sum\limits_{j=0}^{min(i,1/a)} \frac{(-1)^j(1-jx)^i}{j! (i-j)!}$ 。
概率还要除以总测度，即 $x^i$ 。
最终答案： $\sum\limits_{i=0}^\infty \sum\limits_{j=0}^{min(i,1/a)} \frac{(-1)^j}{j! (i-j)!} (\frac{1-jx}{x})^i$ 。
这样其实就可以AC了（无穷取TLE以内的足够，精度要求不高）。
不过我们可以进一步化简，变换求和次序：
$\sum\limits_{j=0}^{1/a} \sum\limits_{i=j}^\infty \frac{(-1)^j}{j! (i-j)!} (\frac{1-jx}{x})^i$ = $\sum\limits_{j=0}^{1/a} \sum\limits_{i=j}^\infty \frac{(-1)^j}{j!} (\frac{1-jx}{x})^j exp(\frac{1-jx}{x})$ 。

Code:

#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i=j;i<=k;++i)
#define per(i,j,k) for(int i=j;i>=k;--i)
#define IOS {cin.tie(0); cout.tie(0); ios::sync_with_stdio(0);}
#define ll long long
#define P pair<int,int>
#define vi vector<int>
#define ls i<<1,l,mid
#define rs i<<1|1,mid+1,r
#define mi int mid=(l+r)>>1;
#define db double
#define ld long double
int T;double x;
const int N=25;
ld fac[N];
int main(){
    IOS
    fac[0]=1;
    rep(i,1,24)fac[i]=i*fac[i-1];
    cin>>T;
    while(T--){
        cin>>x; ld ans=0;
        for(int j=0,f=-1;;++j){
            if(1-j*x<0)break;
            f=-f;
            ans+=f*pow((1-j*x)/x,j)*exp((1-j*x)/x)/fac[j];
        }
        cout<<fixed<<setprecision(10)<<db(ans)<<'\n';
    }
}