#莫队,分块#codevs 6555 洛谷 1494 jzoj 1902 小Z的袜子

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sugar_free_mint/article/details/82049026

题目

有多大的概率抽到两只颜色相同的袜子


分析

首先答案=相同袜子的数量/ C n n 1
但是纯暴力是会超时的,怎么办呢,用莫队+分块愉快地解决问题,注意特判和最简


代码

#include <cstdio>
#include <cmath>
#include <algorithm>
struct rec{int l,r,rk;}e[50001]; long long b1[50001];
int n,m,a[50001],b[50001],ans,a1[50001],t1[50001];
int in(){
    int ans=0; char c=getchar();
    while (c<48||c>57) c=getchar();
    while (c>47&&c<58) ans=ans*10+c-48,c=getchar();
    return ans;
}
bool cmp(const rec &x,const rec &y){
    return b[x.l]==b[y.l]?x.r<y.r:x.l<y.l;
}
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int main(){
    n=in(); m=in(); int t=sqrt(n);
    for (register int i=1;i<=n;i++) a[i]=in(),b[i]=(i-1)/t+1;
    for (register int i=1;i<=m;i++) e[i]=(rec){in(),in(),i};
    std::sort(e+1,e+1+m,cmp); ans+=t1[a[1]]++;
    for (int i=1,l=1,r=1;i<=m;i++){
        while (r<e[i].r) ans+=t1[a[++r]]++;
        while (l>e[i].l) ans+=t1[a[--l]]++;
        while (r>e[i].r) ans-=--t1[a[r--]];
        while (l<e[i].l) ans-=--t1[a[l++]];
        if (!ans) a1[e[i].rk]=0,b1[e[i].rk]=1;
        else {
            a1[e[i].rk]=ans; b1[e[i].rk]=(long long)(e[i].r-e[i].l+1)*(e[i].r-e[i].l)>>1;
            int ggg=gcd(a1[e[i].rk],b1[e[i].rk]);
            a1[e[i].rk]/=ggg; b1[e[i].rk]/=ggg;
        }
    }
    for (int i=1;i<=m;i++)
        printf("%d/%lld\n",a1[i],b1[i]);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/sugar_free_mint/article/details/82049026