题目地址:http://codeforces.com/gym/101911/problem/E
这是一道好的算法组合题,涉及线段树以及双端队列。
-
做完这道题,先写收获:
- 1、差分树状数组(用于区间修改)不能处理区间统一换成数值v的问题(对我而言),只能简单的用作区间加减乘除运算。
- 2、在维护线段树区间的元素的数量的时候,以及其他统计运算的时候,最优思想不应该是将统计元素数量(以及其他需求)整合到树状数组中,而应优先考虑分开思想,即将另一个需求分开来实现。整合思想和数据结构二分思想。
- 3、接2,在统计区间元素数量的时候,双向队列,set(最好不要再构建树了,思考后再使用),栈,map,看看那个容器能符合题目的要求。
- 4、如果线段树只用维护树的最低端,即不需要维护区间和、最大值等区间信息的时候,lazy标记查询到l==r的时候,col[p]就是树的最低端,不需要在构建s[p]的线段树了
AC代码
#include<bits/stdc++.h>
using namespace std;
const int MAX_N = 3e5+50;
int col[4*MAX_N]={0};
int a[MAX_N] = {0};
int n;
int m;
deque<int> dq[MAX_N];
void down(int p)
{
if(col[p]){
col[2*p] = col[p];
col[2*p+1] = col[p];
col[p] = 0;
}
}
void build(int p,int l,int r)
{
if(l==r){
col[p] = a[l];
return ;
}
int mid = (l+r)/2;
build(p*2,l,mid);
build(p*2+1,mid+1,r);
}
void change(int p,int l,int r,int x,int y,int v)
{
if(x<=l&&r<=y){
col[p] = v;
return ;
}
down(p);
int mid = (l+r)/2;
if(x<=mid){
change(p*2,l,mid,x,y,v);
}
if(y>mid){
change(p*2+1,mid+1,r,x,y,v);
}
}
int query(int p,int l,int r,int ind)
{
if(l==r){
return col[p];
}
down(p);
int mid = (l+r)/2;
if(ind<=mid){
return query(p*2,l,mid,ind);
}
else{
return query(p*2+1,mid+1,r,ind);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",a+i);
int k = a[i];
dq[k].push_back(i); ///一开始就排好了
}
build(1,1,n);
scanf("%d",&m);
while(m--){
int x;
scanf("%d",&x);
if(dq[x].size()<=1) continue;
int s = dq[x].front();
int t = dq[x].back();
while(query(1,1,n,s)!=x&&!dq[x].empty()){
dq[x].pop_front();
if(!dq[x].empty()){
s = dq[x].front();
}
}
while(query(1,1,n,t)!=x&&!dq[x].empty()){
dq[x].pop_back();
if(!dq[x].empty()){
t = dq[x].back();
}
}
if(dq[x].size()>1){
change(1,1,n,s,t,x);
}
}
int flag = 1;
for(int i=1;i<=n;i++){
if(!flag) printf(" ");
flag = 0;
printf("%d",query(1,1,n,i));
}
printf("\n");
return 0;
}