HYSBZ - 2038 小Z的袜子(hose) 【莫队】

题目链接:https://www.lydsy.com/JudgeOnline/problem.php?id=2038

思路:

E=\sum C(a[i],2)/C(sum,2)

然后套莫队模板就行了。有个坑点,(R-L+1)*(R-L)会爆int。

代码:

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<set>
using namespace std;
#define MAXN 50005
#define ll long long
int n,m,block;
long long a[MAXN],cnt[MAXN];
long long ans,fz[MAXN],fm[MAXN];
struct query
{
    int l,r,id;
}q[MAXN];
bool cmp(query x,query y)
{
    if(x.l/block!=y.l/block)
        return x.l<y.l;
    return x.r<y.r;
}
void add(long long x)
{
    ans-=cnt[x]*(cnt[x]-1);
    cnt[x]++;
    ans+=cnt[x]*(cnt[x]-1);
}
void move(long long x)
{
    ans-=cnt[x]*(cnt[x]-1);
    cnt[x]--;
    ans+=cnt[x]*(cnt[x]-1);
}
long long gcd(long long a,long long b)
{
    if(b==0) return a;
    return gcd(b,a%b);
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        block=sqrt(n);
        memset(cnt,0,sizeof cnt);
        for(int i=1;i<=n;i++)
            scanf("%lld",&a[i]);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
        }
        sort(q,q+m,cmp);
        int L=1,R=0;
        ans=0;
        for(int i=0;i<m;i++)
        {
            while(R<q[i].r) add(a[++R]);
            while(L>q[i].l) add(a[--L]);
            while(R>q[i].r) move(a[R--]);
            while(L<q[i].l) move(a[L++]);
            if(ans==0)
            {
                fz[q[i].id]=0;
                fm[q[i].id]=1;
                continue;
            }
            ll g=gcd((ll)(R-L+1)*(R-L),ans);
            fz[q[i].id]=ans/g;
            fm[q[i].id]=(ll)(R-L+1)*(R-L)/g;
        }
        for(int i=0;i<m;i++)
            printf("%lld/%lld\n",fz[i],fm[i]);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/u013852115/article/details/81189882