【数据结构】莫队(二)

  今天的内容是带修莫队

例题:P1903 [国家集训队]数颜色 / 维护队列

题目描述

墨墨购买了一套N支彩色画笔(其中有些颜色可能相同),摆成一排,你需要回答墨墨的提问。墨墨会向你发布如下指令:

1、 Q L R代表询问你从第L支画笔到第R支画笔中共有几种不同颜色的画笔。

2、 R P Col 把第P支画笔替换为颜色Col。

为了满足墨墨的要求,你知道你需要干什么了吗?

输入格式

第1行两个整数N,M,分别代表初始画笔的数量以及墨墨会做的事情的个数。

第2行N个整数,分别代表初始画笔排中第i支画笔的颜色。

第3行到第2+M行,每行分别代表墨墨会做的一件事情,格式见题干部分。

输出格式

对于每一个Query的询问,你需要在对应的行中给出一个数字,代表第L支画笔到第R支画笔中共有几种不同颜色的画笔。

输入输出样例

输入 #1
6 5
1 2 3 4 5 5
Q 1 4
Q 2 6
R 1 2
Q 1 4
Q 2 6
输出 #1
4
4
3
4

说明/提示

对于100%的数据,N≤50000,M≤50000,所有的输入数据中出现的所有整数均大于等于1且不超过10^6。

本题可能轻微卡常数

来源:bzoj2120

本题数据为洛谷自造数据,使用CYaRon耗时5分钟完成数据制作。


带修莫队

  莫队是一种很神奇的离线算法。可以解决区间上的问题,通过排序和对询问分块来优化暴力。

  普通的莫队题目都是给定的序列,不断的询问。

  那么如果在询问中夹杂了一些修改操作,莫队还能做吗?

  答案当然是可以的。

  回顾一下莫队的实现方法:

    1.提前预处理

    2.询问分块排序

    3.双指针移动求解。

  唯一会因为修改而产生变化的操作就是3操作。

  可以发现:当出现了修改操作之后,我当前询问处理的区间可能就不是我们本来要找的区间了。

  那么怎么办呢?

  借鉴一下可持久化线段树的思想:时光回溯

  我们可以先把带修莫队当做一般的莫队来做,做完之后如果发现时间对不上就时光倒流。

  准确的说是:考虑两个时间点的操作之间的代价。

  这么说来,只要在普通莫队的一堆 while 循环后面再加两个就OK。

 while(now<q[i].t)
     change(++now);
 while(now>q[i].t)
     change(now--);    

代码

分享图片
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

const int N=50005;
int col[N],n,m,sum[1000005],be[N];
int fans,qnum,cnum,ans[N];

struct qs{
    int l,r,t,id;
}q[N];
struct cs{
    int x,i;
}c[N];

bool cmp(qs x,qs y) {
    if(be[x.l]!=be[y.l])
        return x.l<y.l;
    if(be[x.r]!=be[y.r])
        return x.r<y.r;
    return x.t<y.t;
}

void upd1(int x) {
    if(!sum[col[x]])
        ++fans;
    ++sum[col[x]];
    return;
}

void upd2(int x) {
    --sum[col[x]];
    if(!sum[col[x]])
        --fans;
    return;
}

int l,r,now;

void change(int x) {
    if(c[x].i<=r&&c[x].i>=l) {
        --sum[col[c[x].i]];
        if(!sum[col[c[x].i]])
            --fans;
        if(!sum[c[x].x])
            ++fans;
        ++sum[c[x].x];
    }
    swap(c[x].x,col[c[x].i]);
    return;
}

int main()
{
    cin>>n>>m;
    int xx=pow(n,2.0/3.0);
    char opt; 
    for(int i=1;i<=n;i++) {
        cin>>col[i];
        be[i]=i/xx+1;
    }
    for(int i=1;i<=m;i++) {
        cin>>opt;
        if(opt==Q) {
            ++qnum;
            cin>>q[qnum].l>>q[qnum].r;
            q[qnum].t=cnum;
            q[qnum].id=qnum;
        }
        else {
            ++cnum;
            cin>>c[cnum].i>>c[cnum].x;
        }
    }
    sort(q+1,q+qnum+1,cmp);
    l=r=fans=now=0;
    for(int i=1;i<=qnum;i++) {
        while(l<q[i].l)
            upd2(l++);
        while(l>q[i].l)
            upd1(--l);
        while(r>q[i].r)
            upd2(r--);
        while(r<q[i].r)
            upd1(++r);
        while(now<q[i].t)
            change(++now);
        while(now>q[i].t)
            change(now--);
        ans[q[i].id]=fans;
    }
    for(int i=1;i<=qnum;i++)
        cout<<ans[i]<<endl;
    return 0;
}
数颜色

 


双倍经验 CF940F Machine Learning

  戳这里

相关文章