BZOJ2038 小Z的袜子(hose)

题目链接

---

莫队求区间[L,R]中数字相同的pair数，然后求GCD即可。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define min4(a,b,c,d) min(min(a,b),min(c,d))
#define min3(x,y,z) min(min(x,y),min(y,z))
#define pii make_pair
#define pr pair<int,int>
const int dir[4][2]={0,1,0,-1,1,0,-1,0};
typedef long long ll;
const ll inFF=9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=1e5+5;
int a[maxn],cnt[maxn];
int n,m,u;
ll ans;
struct node
{
    int l,r,id;
}s[maxn];
struct nod
{
    ll x,y;
}res[maxn];
bool cmp(node x,node y){return x.l/u!=y.l/u?x.l<y.l:x.r<y.r;}
void add(int x)
{
    ans+=cnt[a[x]];
    cnt[a[x]]++;
}
void del(int x)
{
    cnt[a[x]]--;
    ans-=cnt[a[x]];
}
int main()
{
    int l,r;
    ll x,y,z;
    scanf("%d %d",&n,&m);
    u=sqrt(n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=m;i++) scanf("%d %d",&s[i].l,&s[i].r),s[i].id=i;
    sort(s+1,s+1+m,cmp);
    l=1,r=0,ans=0;
    for(int i=1;i<=m;i++)
    {
        while(l>s[i].l) add(--l);
        while(l<s[i].l) del(l++);
        while(r<s[i].r) add(++r);
        while(r>s[i].r) del(r--);
        x=ans,y=(ll)(r-l+1)*(r-l)/2;
        z=__gcd(x,y);
        res[s[i].id].x=x/z;
        res[s[i].id].y=y/z;
    }
    for(int i=1;i<=m;i++)
        printf("%lld/%lld\n",res[i].x,res[i].y);
    return 0;
}