Codeforces #341 C. Wet Shark and Flowers 概率论

题目

题目链接：http://codeforces.com/problemset/problem/621/C

题目来源：cf#341

简要题意：$n$个鲨鱼环形排列，每只鲨鱼的数等概率从$[l_i,r_i]$中选取。
$\qquad \quad \ \ \ \ $相邻鲨鱼的数乘积为素数$p$的倍数，出题人就给两个鲨鱼每条$1000$刀，求付出的钱的期望、

题解

由素数的性质可知，$p\mid x\cdot y$则$x,y$至少一个数是$p$的倍数。

单个为$p$的倍数的概率$pos_i=\frac{r_i/p-(l_i-1)/p}{r_i-l_i+1}$

假设当前鲨鱼编号为$x$，左右的鲨鱼编号$l,r$则有单点贡献期望：

$$\begin{align*} exp_x &= (pos_x\cdot (1-pos_l) + pos_x \cdot (1-pos_r) + \frac{pos_x}{2} \cdot (pos_l+pos_r))\cdot 2000\\ &= 2000\cdot pos_x \cdot (2-\frac{pos_l+pos_r}{2}) \end{align*}$$ 前两个为 $x$是$p$倍数而相邻不是，最后一个是 $x$相邻都是的，由于每个位置都要算，所以除以$2$即可。

最终结果就是$\sum\limits_{i=1}^n {exp_i}$。

赛场上由于手误在本题上浪费了大量的时间。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>

#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int> VI;
typedef pair<int,int> PII;
LL powmod(LL a,LL b, LL MOD) {LL res=1;a%=MOD;for(;b;b>>=1){if(b&1)res=res*a%MOD;a=a*a%MOD;}return res;}
// head
const int N = 1e5+5;

double pos[N];

double getPos(int l, int r, int p) {
    int ans = ((r/p) - ((l-1)/p));
    return double(ans) / (r-l+1);
}

double getExp(int x, int n, int p) {
    int l = (x==1 ? n : x-1);// 赛场上此处写成了(x==1 ? n : l-1)，严重手误
    int r = (x==n ? 1 : x+1);// 赛场上此处写成了(x==n ? 1 : r+1)，严重手误
    return pos[x] * (2.0-pos[l]/2-pos[r]/2);
}
int main()
{
    int l, r, n, p;
    double ans = 0.0;
    scanf("%d%d", &n, &p);
    for (int i = 1; i <= n; i++) {
        scanf("%d%d", &l, &r);
        pos[i] = getPos(l, r, p);
    }

    for (int i = 1; i <= n; i++) {
        ans += getExp(i, n, p);
    }
    printf("%.9f\n", ans * 1000.0 * 2);
    return 0;
}