CF1187F Expected Square Beauty

最新推荐文章于 2024-04-30 07:00:00 发布

C202044zxy

最新推荐文章于 2024-04-30 07:00:00 发布

阅读量158

点赞数

分类专栏：数学期望

本文链接：https://blog.csdn.net/C202044zxy/article/details/105427450

版权

数学期望专栏收录该内容

28 篇文章 1 订阅

订阅专栏

一、题目

点此看题

二、解法

设最小子段数为 $B (x)$ ，设 $I_i(x)=[x_i\not=x_{i-1}]$ ，则有这样的关系式：
$B(x)=\sum_{i=1}^nI_i(x)$ 所求是 $E(B(x)^2)$ ，我们来推导一波：
$=E(\sum_{i=1}^nI_i(x)\sum_{j=1}^nI_j(x))$ $=E(\sum_{i=1}^n\sum_{j=1}^nI_i(x)I_j(x))$ $=\sum_{i=1}^n\sum_{j=1}^nE(I_i(x)I_j(x))$ 现在我们来分类讨论一下 $E(I_i(x)I_j(x))$ ：

$i = j$ ，由于 $I_i(x)$ 只能是 $0 / 1$ ，所以它等于 $E(I_i(x))$
$∣ i - j ∣ > 1$ ，由于 $I_i(x)$ 和 $I_j(x)$ 互不影响，所以等于 $E(I_i(x))E(I_j(x))$
$∣ i - j ∣ = 1$ ，设 $q_i=E(x_i=x_{i-1})=\frac{\min(r_i,r_{i-1})-\max(l_i.l_{i-1})}{(r_i-l_i)(r_{i-1}l_{i-1})}$ ，我们要求的是 $E(x_{i-1}\not=x_{i}\&x_{i}\not=x_{i+1})$ ，利用简单的容斥原理，可以得出所求为 $1-q_i-q_{i+1}-E(x_{i-1}=x_{i}\&x_i=x_{i+1})$ ，最后一项利用类似的方法算即可。

不难发现总时间复杂度是 $O (n)$ 的，贴个代码

#include <cstdio>
#include <iostream>
using namespace std;
#define int long long
const int M = 200005;
const int jzm = 1e9+7;
int read()
{
    int x=0,flag=1;
    char c;
    while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*flag;
}
int n,ans,sum,l[M],r[M],q[M],E[M];
int qkpow(int a,int b)
{
    int r=1;
    while(b>0)
    {
        if(b&1) r=r*a%jzm;
        a=a*a%jzm;
        b>>=1;
    }
    return r;
}
int calc(int i)
{
    int ret=0;
    if(i>1)
    {
        int t=min(r[i],min(r[i-1],r[i+1]))-max(l[i],max(l[i-1],l[i+1]));
        ret=max(0ll,t*qkpow((r[i]-l[i])*(r[i-1]-l[i-1])%jzm*(r[i+1]-l[i+1])%jzm,jzm-2)%jzm);
    }
    return (1-q[i]-q[i+1]+ret)%jzm;
}
signed main()
{
    n=read();
    for(int i=1;i<=n;i++)
        l[i]=read();
    for(int i=1;i<=n;i++)
    {
        r[i]=read()+1;
        int R=min(r[i],r[i-1]),L=max(l[i],l[i-1]);
        q[i]=max(0ll,(R-L)*qkpow((r[i]-l[i])*(r[i-1]-l[i-1])%jzm,jzm-2)%jzm);
        E[i]=(1-q[i]+jzm)%jzm;
        sum=(sum+E[i])%jzm;
    }
    for(int i=1;i<=n;i++)
    {
        int tmp=sum;
        for(int j=max(1ll,i-1);j<=min(n,i+1);j++)
            tmp=(tmp-E[j])%jzm;
        ans=(ans+E[i]*tmp+E[i])%jzm;
        if(i>1) ans=(ans+calc(i-1))%jzm;
        if(i<n) ans=(ans+calc(i))%jzm;
    }
    printf("%d\n",(ans+jzm)%jzm);
}