Problem E Painting the Fence (区间染色)

https://codeforces.com/gym/101911/problem/E

题意

n个数，每个数都代表一种颜色，m个操作，每个操作给出一个颜色col，要求将[最左边的col,最右边的col]这个区间都染成col。输出最终序列的颜色。

题解

关键点: 每种颜色只需要操作一次!!!

因为每种颜色第一次绝对是覆盖最多的，下次同样的操作不可能再覆盖更多的地方。 所以一旦覆盖，这个区间其实就可以缩成一个点来代替了。

set<int> g[i]: 记录颜色i的所有位置。

对于已经操作过的颜色直接跳过，否则直接找到最左最右，将这区间内的颜色的位置在对应的颜色set里删除。然后记录下这个颜色覆盖的区间即可。即nxt[l] = r+1。

最后遍历1~n，对于[l,nxt[l])之间的输出一个颜色即可。
每个位置只会操作一次，时间复杂度$O(nlogn)$

代码

#include <bits/stdc++.h>
using namespace std;
#define FOR0(a,b) for(int i = a; i < b; ++i)
#define FORE(a,b) for(int i = a; i <= b; ++i)
typedef long long ll;
typedef pair<int,int> pii;

const int maxn = 3e5+5;
set<int> g[maxn];
int nxt[maxn];
int n;
int a[maxn], vis[maxn];

void del(int l,int r) {
	for(int i = l+1; i <= r; i = nxt[i]) {
		g[a[i]].erase(i);
	}
}
int main() {
	scanf("%d", &n);
	int col;
	for(int i = 0; i < n; ++i) {
		scanf("%d", &a[i]);
		g[a[i]].insert(i);
		nxt[i] = i+1;
	}

	int m;
	scanf("%d", &m);
	for(int i = 0; i < m; ++i) { 
		scanf("%d", &col);
		if(vis[col] || g[col].size() < 2) continue;
		vis[col] = 1;
		int l = *(g[col].begin());
		int r = *(g[col].rbegin());
		del(l,r);
		nxt[l] = r+1;
	}

	for(int i = 0; i < n; i = nxt[i]) {
		for(int j = i; j < nxt[i]; ++j)
			printf("%d ",a[i]);
	}
	puts("");
	return 0;
}