洛谷 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復制
4 4 3 4說明/提示
對于30%的數據,n,m≤10000
對于60%的數據,n,m≤50000
對于所有數據,n,m≤133333
所有的輸入數據中出現的所有整數均大于等于1且不超過10^6。
本題可能輕微卡常數
題解:
題意很明確,其實就是個莫隊單點修改的模板題
莫隊的查詢操作+修改操作
前面都是套上莫隊的板子,與莫隊不同的是要用一個變量來記錄當前進行幾次修改了
對于查詢操作,如果當前改的比本次查詢需要改的少,就改過去
反之如果改多了就改回來
說的聽繞口的
比如,我們現在已經進行了3次修改,本次查詢是在第5次修改之后,那我們就執行第4,5次修改
如果我們進行了第三次修改,但是本次查詢實在第1次,多修改的第2,3次就要改回來
然后修改后的顏色和原本的顏色要看看對結果是否產生影響
代碼:
代碼有詳細解釋
#include<cstdio> #include<cmath> #include<algorithm> using namespace std; const int MAXN=2*1e6+10; inline int read() {char c=getchar();int x=0,f=1;while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}while(c>='0'&&c<='9'){x=x*10+c-'0',c=getchar();}return x*f; } int N,M; int a[MAXN],where[MAXN]; struct Query {int x,y,pre,id; }Q[MAXN]; int Qnum=0; struct Change {int pos,val; }C[MAXN]; int Cnum=0; int color[MAXN],ans=0,base,out[MAXN]; int comp(const Query &a,const Query &b) {if(a.x!=b.x) return where[a.x]<where[b.x];if(a.y!=b.y) return where[a.y]<where[b.y];return a.pre<b.pre; } void Add(int val) { if(++color[val]==1) ans++; } void Delet(int val) {if(--color[val]==0) ans--; } void Change(int now,int i) {if(C[now].pos>=Q[i].x&&C[now].pos<=Q[i].y)//注意:只有修改在查詢的區間內才會對查詢的結果產生影響 {if( --color[a[C[now].pos]] == 0 ) ans--;//如果換后,原本的顏色消失,且再無此顏色 if( ++color[C[now].val] == 1) ans++; //如果換后,出現新顏色 }swap(C[now].val,a[C[now].pos]);//這里有個很巧妙的操作//對于一個操作,下一次需要為的顏色是本次被改變的顏色//比如,我把顏色3改為了7,那么再進行這次修改的時候就是把7改為3//所以直接交換兩種顏色就好 //修改該位置->還原該位置->修改該位置...如此循環 } void MoQueue() {int l=1,r=0,now=0; for(int i=1;i<=Qnum;i++){while(l<Q[i].x) Delet(a[l++]);while(l>Q[i].x) Add(a[--l]);while(r<Q[i].y) Add(a[++r]);while(r>Q[i].y) Delet(a[r--]);//以上四句為莫隊模板 while(now<Q[i].pre) now++,Change(now,i);//改少了,改過去 while(now>Q[i].pre) Change(now,i),now--;//改多了,改回來 out[Q[i].id]=ans;//統計答案 }for(int i=1;i<=Qnum;i++)printf("%d\n",out[i]); } int main() {N=read();M=read();base=sqrt(N);for(int i=1;i<=N;i++) a[i]=read(),where[i]=(i-1)/base+1;while(M--){char opt[5];scanf("%s",opt);if(opt[0]=='Q'){Q[++Qnum].x=read();Q[Qnum].y=read();Q[Qnum].pre=Cnum;//別忘了記錄最近的修改位置 Q[Qnum].id=Qnum; }else if(opt[0]=='R'){C[++Cnum].pos=read();C[Cnum].val=read();}}sort(Q+1,Q+Qnum+1,comp);//玄學排序 MoQueue();return 0; }上面這個代碼有部分點超時
下面這個AC了
總結
以上是生活随笔為你收集整理的洛谷 P1903 [国家集训队]数颜色 / 维护队列的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python实现网关_用python实现
- 下一篇: Graph Coloring I