BZOJ[2038][2009国家集训队]小Z的袜子(hose) 莫队

最新推荐文章于 2021-09-18 17:33:27 发布

Duan2baka

最新推荐文章于 2021-09-18 17:33:27 发布

阅读量255

点赞数

分类专栏： BZOJ 莫队

本文链接：https://blog.csdn.net/WADuan2/article/details/79056281

版权

BZOJ 同时被 2 个专栏收录

118 篇文章 0 订阅

订阅专栏

莫队

3 篇文章 0 订阅

订阅专栏

传送门ber~

要求区间 $[l,r]$ 内选出两个相同颜色的概率，即求

(C 2 c n t 1 + C 2 c n t 2 + . . . + C 2 c n t k) / C 2 r - l + 1

$(C^{2}_{cnt_1}+C^{2}_{cnt_2}+...+C^{2}_{cnt_k})/C^{2}_{r-l+1}$

( c n t 2 1 - c n t 1 + c n t 2 2 - c n t 2 + . . . + c n t 2 k - c n t k ) 2 / ( r - l + 1 ) * ( r - l ) 2

$\frac {(cnt_1^2-cnt_1+cnt_2^2-cnt_2+...+cnt_k^2-cnt_k)}{2}/\frac {(r-l+1)*(r-l)}{2}$
其中

cnti $cnt_i$ 为颜色为

i $i$ 的袜子有多少个，所以

c n t 1 + c n t 2 + c n t 3 + . . . + c n t k = r - l + 1

$cnt_1+cnt_2+cnt_3+...+cnt_k=r-l+1$
所以刚才的式子就化为

(c n t 21 + c n t 22 + . . . + c n t 2 k - (r - l + 1)) / ((r - l + 1) * (r - l))

$(cnt_1^2+cnt_2^2+...+cnt_k^2-(r-l+1))/((r-l+1)*(r-l))$
区间平方和可以用莫队维护，见 BZOJ[3781]

代码如下：

#include<algorithm>
#include<ctype.h>
#include<cstdio>
#include<cmath>
#define N 50050
using namespace std;
inline int read(){
    int x=0,f=1;char c;
    do c=getchar(),f=c=='-'?-1:f; while(!isdigit(c));
    do x=(x<<3)+(x<<1)+c-'0',c=getchar(); while(isdigit(c));
    return x*f;
}
typedef long long LL;
int n,m,Block_size,l=1,r;
int a[N],block[N],cnt[N];
LL tmp;
struct Result{
    LL x,y;
    Result(){}
    Result(int _,int __):x(_),y(__){}
    inline void print(){
        printf("%lld/%lld\n",x,y);
    }
}ans[N];
struct Query{
    int l,r,id;
}q[N];
inline bool cmp(Query a,Query b){
    return block[a.l]==block[b.l]?a.r<b.r:block[a.l]<block[b.l];
}
LL gcd(LL x,LL y){
    return (!y)?x:gcd(y,x%y);
}
inline Result Trans(LL x,LL y){
    LL t=gcd(x,y);
    return Result(x/t,y/t);
}
inline void Update(int x,int k){
    tmp-=cnt[x]*cnt[x];
    cnt[x]+=k;
    tmp+=cnt[x]*cnt[x];
}
int main(){
    n=read();m=read();
    Block_size=sqrt(n);
    for(int i=1;i<=n;i++){
        a[i]=read();
        block[i]=(i-1)/Block_size+1;
    }
    for(int i=1;i<=m;i++){
        q[i].l=read();q[i].r=read();
        q[i].id=i;
    }
    sort(q+1,q+m+1,cmp);
    for(int i=1;i<=m;i++){
        while(l<q[i].l) Update(a[l++],-1);
        while(l>q[i].l) Update(a[--l],1);
        while(r<q[i].r) Update(a[++r],1);
        while(r>q[i].r) Update(a[r--],-1);
        if(q[i].r==q[i].l) ans[q[i].id].y=1;
        else ans[q[i].id]=Trans(tmp-(q[i].r-q[i].l+1),1ll*(q[i].r-q[i].l+1)*(q[i].r-q[i].l));
    }
    for(int i=1;i<=m;i++)
        ans[i].print();
return 0;
}