bzoj 1293: [SCOI2009]生日礼物 STL

→←

对于每个彩珠我们保存它的下标和颜色，然后按下标排序

之后用set存对于每个颜色我们选中的彩珠的下标

如果insert的时候碰到set的size为0，说明这是一个新的颜色，我们就令difcol++

每次使右端点向右移动1

然后如果左端点的彩珠的颜色的set的size大于1，说明这个彩珠我们不需要了，就从set中erase掉

不用考虑相同坐标中的重复彩珠，因为对于一个坐标不论有多少个相同颜色的彩珠我们都是只选一个就够了

删除彩珠时也不用必须一次性把一个坐标上的彩珠全都删掉，如果删不掉说明我们一定不能删掉，删掉了那就删掉了，没有影响

最后如果difcol==k，就令ans与右端点坐标-左端点坐标取一个min

代码：

#include<cstdio>
#include<set>
#include<algorithm>
 
#define ll long long
#define inf (1<<31)-1
 
using namespace std;

struct node{
	int col,x;
	friend bool operator<(node a,node b){return a.x<b.x;};
};

int n,m;
int difcol=0;
set<int>num[66];
node a[1000010];
ll ans=inf;

int main(){
	scanf("%d%d",&n,&m);
	int tot=0;
	for(int i=1; i<=m; i++){
		int x;
		scanf("%d",&x);
		for(int j=0; j<x; j++){
			scanf("%d",&a[tot].x);
			a[tot].col=i;
			tot++;
		}
	}
	sort(a,a+n);
	int l=0,r=0;
	while(r<n){
		if(num[a[r].col].size()==0)difcol++;
		num[a[r].col].insert(a[r].x),r++;
		while(num[a[l].col].size()>1)num[a[l].col].erase(a[l].x),l++;
		if(difcol==m)ans=min(ans,(ll)a[r-1].x-a[l].x);
	}
	printf("%d\n",ans);
	
    return 0;
}