bzoj2038-小z的袜子莫队算法

Posted 2020-12-05 konjakjuruo

莫队例题。

莫队学习：https://www.cnblogs.com/Paul-Guderian/p/6933799.html

本题 分子是sigma(c(sum[a[i]],2))，分母是sigma(l-r+1,2); 维护分子和即可。

莫队适用范围：离线，区间，区间转移到下一格O(1)。

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<cstring>
 4 #include<iostream>
 5 #include<algorithm>
 6 #include<cmath>
 7 using namespace std;
 8 
 9 const int N=50010;
10 typedef long long LL;
11 int n,m,sq,a[N];
12 LL ans,gcd,sum[N];
13 struct node{
14     int l,r,k,id;
15     LL fz,fm;
16 };
17 node b[N];
18 
19 int maxx(int x,int y){return x>y?x:y;}
20 bool cmp1(node x,node y)
21 {
22     if(x.k==y.k) return x.r<y.r;
23     return x.l<y.l;
24 }
25 bool cmp2(node x,node y)
26 {
27     return x.id<y.id;
28 }
29 void revise(int x,int d)
30 {
31     ans-=sum[a[x]]*(sum[a[x]]-1)/2;
32     sum[a[x]]+=d;
33     ans+=sum[a[x]]*(sum[a[x]]-1)/2;
34 }
35 LL getgcd(LL a,LL b)
36 {
37     return b?getgcd(b,a%b):a;
38 }
39 
40 int main()
41 {
42     freopen("a.in","r",stdin);
43     freopen("a.out","w",stdout);
44     scanf("%d%d",&n,&m);sq=sqrt(n);
45     for(int i=1;i<=n;i++) scanf("%d",&a[i]);
46     for(int i=1;i<=m;i++)
47     {
48         scanf("%d%d",&b[i].l,&b[i].r);
49         b[i].k=b[i].l/sq;
50         b[i].id=i;
51     }
52     sort(b+1,b+m+1,cmp1);
53     memset(sum,0,sizeof(sum));
54     int l=1,r=1;sum[a[1]]=1;ans=0;
55     for(int i=1;i<=m;i++)
56     {
57         while(l<b[i].l) {revise(l,-1);l++;}
58         while(l>b[i].l) {revise(l-1,1);l--;}
59         while(r<b[i].r) {revise(r+1,1);r++;}
60         while(r>b[i].r) {revise(r,-1);r--;}
61         b[i].fz=ans;b[i].fm=((LL)(b[i].r-b[i].l+1))*((LL)(b[i].r-b[i].l))/2;
62         gcd=getgcd(b[i].fz,b[i].fm);
63         b[i].fz/=gcd;b[i].fm/=gcd;
64         if(b[i].fz==0) b[i].fm=1;
65     }
66     sort(b+1,b+m+1,cmp2);
67     for(int i=1;i<=m;i++)
68     {
69         printf("%lld/%lld
",b[i].fz,b[i].fm);
70     }
71     return 0;
72 }