【BZOJ 2038】小Z的袜子

【题目链接】

https://www.lydsy.com/JudgeOnline/problem.php?id=2038

【算法】

莫队算法

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN 50010
typedef long long ll;

struct info
{
        ll x,y;
} ans[MAXN];
struct Query
{
        ll l,r,id;
} q[MAXN],tmp[MAXN];

ll i,j,n,m,len,block,t,sum,l,r,k,g;
ll a[MAXN],num[MAXN];

inline ll gcd(ll x,ll y)
{
        return y ==  ? x : gcd(y,x%y);
}
inline bool cmp1(Query a,Query b)
{
        return a.l < b.l;
}
inline bool cmp2(Query a,Query b)
{
        return a.r < b.r;
}
inline void add(ll l,ll r,ll val)
{
        ll i;
        for (i = l; i <= r; i++)
        {
                sum -= num[a[i]] * (num[a[i]] - );
                num[a[i]] += val;
                sum += num[a[i]] * (num[a[i]] - );
        }
}

int main()
{

        scanf("%lld%lld",&n,&m);
        for (i = ; i <= n; i++) scanf("%lld",&a[i]);
        for (i = ; i <= m; i++)
        {
                scanf("%lld%lld",&q[i].l,&q[i].r);
                q[i].id = i;
        }
        sort(q+,q+m+,cmp1);
        len = sqrt(n);
        block = n / len + (n % len != );
        j = ;
        for (i = ; i <= block; i++)
        {
                t = ; sum = ;
                while (j <= m && q[j].l > (i - ) * len && q[j].l <= i * len)
                {
                        tmp[++t] = q[j];
                        j++;
                }
                memset(num,,sizeof(num));
                sort(tmp+,tmp+t+,cmp2);
                l = tmp[].l; r = tmp[].l - ;
                for (k = ; k <= t; k++)
                {
                        if (l < tmp[k].l) add(l,tmp[k].l-,-);
                        else if (l > tmp[k].l) add(tmp[k].l,l-,);
                        add(r+,tmp[k].r,);
                        l = tmp[k].l; r = tmp[k].r;
                        ans[tmp[k].id].x = sum;
                        ans[tmp[k].id].y = (tmp[k].r - tmp[k].l + ) * (tmp[k].r - tmp[k].l);
                }
        }
        for (i = ; i <= m; i++)
        {
                if (!ans[i].x) printf("0/1\n");
                else
                {
                        g = gcd(ans[i].x,ans[i].y);
                        ans[i].x /= g;
                        ans[i].y /= g;
                        printf("%lld/%lld\n",ans[i].x,ans[i].y);
                }
        }
        return ;

}