【数据结构】莫队(二)

前端之家收集整理的这篇文章主要介绍了【数据结构】莫队(二)前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

  今天的内容带修莫队

例题: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
  1. 6 5
  2. 1 2 3 4 5 5
  3. Q 1 4
  4. Q 2 6
  5. R 1 2
  6. Q 1 4
  7. Q 2 6
输出 #1
  1. 4
  2. 4
  3. 3
  4. 4

说明/提示

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

本题可能轻微卡常数

来源:bzoj2120

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


带修莫队

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

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

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

  答案当然是可以的。

  回顾一下莫队的实现方法

    1.提前预处理

    2.询问分块排序

    3.双指针移动求解。

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

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

  那么怎么办呢?

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

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

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

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

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

代码

分享图片

  1. #include<iostream>
  2. #include<cstdio>
  3. #include<cstring>
  4. #include<algorithm>
  5. #include<cmath>
  6. using namespace std;
  7. const int N=50005;
  8. int col[N],n,m,sum[1000005],be[N];
  9. int fans,qnum,cnum,ans[N];
  10. struct qs{
  11. int l,r,t,id;
  12. }q[N];
  13. struct cs{
  14. int x,i;
  15. }c[N];
  16. bool cmp(qs x,qs y) {
  17. if(be[x.l]!=be[y.l])
  18. return x.l<y.l;
  19. if(be[x.r]!=be[y.r])
  20. return x.r<y.r;
  21. return x.t<y.t;
  22. }
  23. void upd1(int x) {
  24. if(!sum[col[x]])
  25. ++fans;
  26. ++sum[col[x]];
  27. return;
  28. }
  29. void upd2(int x) {
  30. --sum[col[x]];
  31. if(!sum[col[x]])
  32. --fans;
  33. return;
  34. }
  35. int l,now;
  36. void change(int x) {
  37. if(c[x].i<=r&&c[x].i>=l) {
  38. --sum[col[c[x].i]];
  39. if(!sum[col[c[x].i]])
  40. --fans;
  41. if(!sum[c[x].x])
  42. ++fans;
  43. ++sum[c[x].x];
  44. }
  45. swap(c[x].x,col[c[x].i]);
  46. return;
  47. }
  48. int main()
  49. {
  50. cin>>n>>m;
  51. int xx=pow(n,2.0/3.0);
  52. char opt;
  53. for(int i=1;i<=n;i++) {
  54. cin>>col[i];
  55. be[i]=i/xx+1;
  56. }
  57. for(int i=1;i<=m;i++) {
  58. cin>>opt;
  59. if(opt==Q) {
  60. ++qnum;
  61. cin>>q[qnum].l>>q[qnum].r;
  62. q[qnum].t=cnum;
  63. q[qnum].id=qnum;
  64. }
  65. else {
  66. ++cnum;
  67. cin>>c[cnum].i>>c[cnum].x;
  68. }
  69. }
  70. sort(q+1,q+qnum+1,cmp);
  71. l=r=fans=now=0;
  72. for(int i=1;i<=qnum;i++) {
  73. while(l<q[i].l)
  74. upd2(l++);
  75. while(l>q[i].l)
  76. upd1(--l);
  77. while(r>q[i].r)
  78. upd2(r--);
  79. while(r<q[i].r)
  80. upd1(++r);
  81. while(now<q[i].t)
  82. change(++now);
  83. while(now>q[i].t)
  84. change(now--);
  85. ans[q[i].id]=fans;
  86. }
  87. for(int i=1;i<=qnum;i++)
  88. cout<<ans[i]<<endl;
  89. return 0;
  90. }
数颜色

 


双倍经验 CF940F Machine Learning

  戳这里

猜你在找的数据结构相关文章