逛画展——尺取/单调队列

逛画展

逛画展 - 计蒜客 (jisuanke.com)

博览馆正在展出由世上最佳的 M 位画家所画的图画。蒜头君想到博览馆去看这几位大师的作品，可是那里的博览馆有一个很奇怪的规定，就是在购买门票时必须说明两个数字 a 和 b，代表他要看展览中的第 a 幅至第 b 幅画（包含 a 和 b）之间的所有图画，而门票的价钱就是一张画一元。

为了看到更多名师的画，蒜头君希望入场后可以看到所有名师的图画（至少各一张），可是他又想节省金钱。作为wangjy的朋友，他请你写一个程序决定他购买门票时的 a 值和 b 值。

输入格式

第一行是 N (1≤N≤106) 和 M (1≤M≤2000)，分别代表博览馆内的图画总数及这些图画是由多少位名师的画所绘画的。

其后的一行包含 N 个数字，它们都介于 11 和 M 之间，代表该位名师的编号。

输出格式

a 和 b，由一个空格符所隔开。保证有解，如果多解，输出a最小的。

分析：

读完题我们很容易可以想到这道题可以用尺取法来确定范围a~b，那么我们可以用两个数组来分别表示图画状态以及画展画家序号。其中da代表当前画家图画状态，下标代表画家序号，初始值都为0代表在a~b范围内尚未有该画家作品。

下面贴上代码：

#include<algorithm>
#include<iostream>
using namespace std;
const int N=1e6+5;
int da[N]={0},a[N]={0};
int main()
{
	
	int num=0,ans,x,y;
	int n,m;
	scanf("%d %d",&n,&m);
	ans=n;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	int l=1,r=1; 	
	while(l<=r&&r<n)
	{
		while(num<m&&r<=n)
		{
			if(da[a[r]]==0)//判断该画家的作品是否被录入过
				num++;
			da[a[r++]]++;
		} 
		if(num<m)
			break;
		if(r-l<ans)//在符合前提num<m情况下 如果找到更短的范围则更新
		{
			ans=r-l;
			x=l,y=r-1;
		}	
		if(da[a[l++]]==1)//删去最左侧的作品判断是否会导致看过的画家数减少
			num--;
		da[a[l-1]]--;	
	}
	printf("%d %d",x,y);
	return 0;
}

由上述双指针写法可以联想到队列，只不过将利用左支针和右指针处理数据变为处理队头和队尾数据。所以这里用了双向队列更方便处理。这里同样用了和上一种方法相同的两个数组存储信息，需要注意的是，每录入一个数据都要判断它的作者是否与队头的作者重复，重复则删去队头作品，这时队头数据更新，我们还要判断更新后的队头作品作者是否出现超过1次，超过则去除，直到队头作品作者只出现一次。

下面贴上代码：

#include<algorithm>
#include<deque>
#include<iostream>
using namespace std;
const int N=1e6+5;
int da[N]={0},a[N]={0};
struct node
{
	int id;//作品序号
	int wri;//作品作者
}p;
int main()
{
	
	int num=0,x=1,y=N;
	deque<node>q;
	int n,m;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);	
	for(int i=1;i<=n;i++)
	{
		p.id=i,p.wri=a[i];
		q.push_back(p);
		if(da[a[i]]==0)
			num++;
		da[a[i]]++;	
		while(!q.empty()>1&&q.front().wri==q.back().wri)
		{
			da[q.front().wri]--;
			q.pop_front();
		}
		while(da[q.front().wri]>1)
		{
			da[q.front().wri]--;
			q.pop_front();
		}	
		if(num==m)
		{
			int t=q.back().id-q.front().id;
			if((y-x)>t)
				x=q.front().id,y=q.back().id;
		}	
	}
	printf("%d %d",x,y);
	return 0;
}