【莫队算法】的小例题

【BZOJ 2038】【2009 国家集训队】

具体来说，小Z把这N只袜子从1到N编号，然后从编号L到R(L 尽管小Z并不在意两只袜子是不是完整的一双，甚至不在意两只袜子是否一左一右，他却很在意袜子的颜色，毕竟穿两只不同色的袜子会很尴尬。
你的任务便是告诉小Z，他有多大的概率抽到两只颜色相同的袜子。当然，小Z希望这个概率尽量高，所以他可能会询问多个(L,R)以方便自己选择。
【题解】
对于L,R的询问。设其中颜色为x,y,z….的袜子的个数为a,b,c。。。
那么答案即为$\frac{a*(a-1)/2+b*(b-1)/2+c*(c-1)/2....}{(R-L+1)*(R-L)/2}$
化简$\frac{a^2+b^2+c^2+...x^2-(a+b+c+d+.....)}{(R-L+1)*(R-L)}$
即$\frac{a^2+b^2+c^2+...x^2-(R-L+1)}{(R-L+1)*(R-L)}$
所以这道题目的关键是求一个区间内每种颜色数目的平方和。

莫队直接搞
【代码】

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

#define N 50010
using namespace std;
typedef long long ll;

int c[N],pos[N];
ll num[N],up[N],dw[N],ans,aa,bb,cc;
struct node{int l,r,id;}qu[N];
bool cmp(node a,node b) {
    return pos[a.l] == pos[b.l] ? a.r < b.r : pos[a.l] < pos[b.l];
}
ll gcd(ll a,ll b) { return b == 0 ? a : gcd(b,a%b); }

void update(int x,int d)
{
    ans -= num[c[x]]*num[c[x]];
    num[c[x]]+=d;
    ans += num[c[x]]*num[c[x]];
}

int main()
{
    int n,q,bs,pl,pr,id;
    scanf("%d%d",&n,&q);
    bs = (int)sqrt(n);
    for(int i = 1;i <= n;i++) {
        scanf("%d",&c[i]);
        pos[i] = (i-1)/bs;
    }
    for(int i = 1;i <= q;i++) {
        scanf("%d%d",&qu[i].l,&qu[i].r);
        qu[i].id = i;
    }

    memset(num,0,sizeof(num));
    sort(qu+1,qu+q+1,cmp);
    pl = 1;pr = 0;
    ans = 0;
    for(int i = 1;i <= q;i++) {
        id = qu[i].id;
        if(qu[i].l == qu[i].r) {up[id] = 0,dw[id] = 1;continue;}
        if(pr < qu[i].r)
            for(int j = pr+1;j <= qu[i].r;j++)
                update(j,1);
        else
            for(int j = pr;j > qu[i].r;j--)
                update(j,-1);
        pr = qu[i].r;
        if(pl < qu[i].l)
            for(int j = pl;j < qu[i].l;j++)
                update(j,-1);
        else
            for(int j = pl-1;j >= qu[i].l;j--)
                update(j,1);

        pl = qu[i].l;
        aa = ans - qu[i].r + qu[i].l - 1;
        bb = (ll)(qu[i].r-qu[i].l+1)*(qu[i].r-qu[i].l);
        cc = gcd(aa,bb);
        aa /= cc; bb /= cc;
        up[id] = aa,dw[id] = bb;
    }
    for(int i = 1;i <= q;i++)
        printf("%lld/%lld\n",up[i],dw[i]);
    return 0;
}

【HDU 4638】Group

传送门
给你一个长度为n(1<=n<=100000)的数列。数列中的值互不相同且1<=ai<=n。对于一个给定的区间。[L,R]ans就为下标[L,R]的数中值域连续的块的块数。
【题解】
用莫队来维护，用一个数组vis来判断一个数值x有没有出现，再通过x-1和x+1来判断块数目的加减。
【代码】

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

#define N 100010
using namespace std;

int ans,a[N],o[N],pos[N];
bool vis[N];
int bs;
struct node{int l,r,id;} qu[N];
bool cmp(node a,node b){return pos[a.l]==pos[b.l] ? a.r<b.r : pos[a.l]<pos[b.l];}

void update(int x,bool d)
{
    vis[x] = d;
    if(d) ans += 1-vis[x-1]-vis[x+1];
    else ans += vis[x-1]+vis[x+1]-1;
}

int main()
{
    int T,n,m;
    scanf("%d",&T);
    while(T--) {
        scanf("%d%d",&n,&m);
        bs = ceil(sqrt(n));
        for(int i = 1;i <= n;i++) {
            scanf("%d",&a[i]);
            pos[i] = (i-1)/bs;
        }
        for(int i = 1;i <= m;i++) {
            scanf("%d%d",&qu[i].l,&qu[i].r);
            qu[i].id = i;
        }
        sort(qu+1,qu+m+1,cmp);
        memset(vis,0,sizeof(vis));

        int pl = 1,pr = 0;
        ans = 0;
        for(int i = 1;i <= m;i++) {
            int x = qu[i].id;
            if(pr < qu[i].r)
                for(int j = pr+1;j <= qu[i].r;j++) update(a[j],1);
            else
                for(int j = pr;j > qu[i].r;j--) update(a[j],0);
            if(pl > qu[i].l)
                for(int j = pl-1;j >= qu[i].l;j--) update(a[j],1);
            else
                for(int j = pl;j < qu[i].l;j++) update(a[j],0);
            pr = qu[i].r; pl = qu[i].l;
            o[x] = ans;
        }
        for(int i = 1;i <= m;i++) printf("%d\n",o[i]);
    }
    return 0;
}