P2564 [SCOI2009]生日礼物（队列，双指针，模拟）

题目链接

https://www.luogu.org/problem/P2564

题目大意

给定N个彩珠的位置，求容纳所有类型彩珠的最短区间长度

思路

双指针，分为维护区间开始和区间结束的位置，同时统计这一段区间内首次出现的彩珠类型，当区间内不同类型的彩珠数量达到给定值k时，记录并更新答案（区间长度）

下一步移动头指针，如果区间内该彩珠类型为0了，那么减少cnt的数量

有点类似于求一个最小长度的移动窗口使得该窗口包含所以类型的珍珠，但不要想复杂了QAQ…

#include
using namespace std;
#define N 1000010
int n,k;
int tot=0;
struct node{
	int col;
	int x;
	bool operator <(const node &b) const{
		if(x==b.x) return col<b.col;
		return x<b.x;
	}
}a[N];
int Q[N];
int vis[N];
const int inf=0x7fffffff;

//双指针统计模拟
int solve(){
	int ans=inf;
	int head=0,tail=0;
	int cnt=0;
	while(head<=tail && tail<=n){
		// printf("%d %d\n",head,tail);
		while(tail!=n && cnt<k){
			tail++;
			if(!vis[a[tail].col]) cnt++;

			vis[a[tail].col]++;
		}

		if(cnt>=k)
			ans=min(ans,a[tail].x-a[head+1].x);
		else 
			break;

		head++;
		vis[a[head].col]--;
		if(!vis[a[head].col]) cnt--;
	}

	return ans;
}
int main(){
	// freopen("p.txt","r",stdin);
	scanf("%d%d",&n,&k);
	// int R=0,L=(1<<31)-1;
	for(int i=1;i<=k;i++){
		int num;scanf("%d",&num);
		for(int j=1;j<=num;j++){
			int x;
			scanf("%d",&x);
			a[++tot]=node{i,x};
		}
	}
	sort(a+1,a+tot+1);//1~tot

	int ans=solve();
	printf("%d\n",ans);
	return 0;
}