Cow Lineup S——离散化、单调队列、双指针

题目描述

思路

• x、id不大于1亿，数据量太大，使用离散化
• 将id离散化成一串从1开始连续的编号，使用map集合进行离散化
• 使用双指针维护一段区间，这段区间满足每个编号都包含

如何使用map集合进行离散化？

• 维护一个变量nums，从1开始，每个编号的牛第一次被遍历的时候将nums作为新的编号，nums++
• 使用map集合判断这个牛的编号是不是第一次遍历， key值表示原编号，value表示新的编号
• 用一个结构体存储牛的信息：位置和编号，按位置进行排序，并且进行离散化的时候，将编号进行更新
• 定义一个队列，队列中存储每头牛的信息，将新的牛从队尾插入，并且记录这头牛的编号是否是第一次入列，当每个编号的牛都入列了，当前区间满足条件，进行比较
• 当最后一头牛入列并且队头存储的牛的编号只有一种时，退出循环

代码实现

#include <iostream>
#include <map>
#include <algorithm>

using namespace std;

const int N = 50010;
map<int, int> a;	// 将ID编号离散化成一串连续的编号
int c[N];	// 下标为编号，出现了多少次
int nums;	// 牛的种类
int n;

struct node
{
	int x, id;
}niu[N], q[N];	// niu表示每头牛的信息：位置和编号	
				// q表示队列，存储牛的信息

bool cmp(node a, node b)
{
	return a.x < b.x;
}

int main()
{
	cin >> n;
	for(int i = 0;i < n;i ++)
	{
		int id;
		cin >> niu[i].x >> id;
		if(!a.count(id))	// 如果当前编号没有出现过，牛的种类加1
		{
			nums++;
			a[id] = nums;
		}
		niu[i].id = a[id]; // 将编号更新成新的编号
	}
	sort(niu, niu + n, cmp);	// 按位置进行排序
	int hh = 0, tt = 0;	// hh队头、tt队尾
	int res = 0;	// 表示当前队列中有多少不同编号的牛
	q[tt] = {niu[0].x, niu[0].id};	// 先将第一头牛的信息加入到队列中
	res++;
	int ress = 0x3f3f3f3f;	// 满足条件的区间长度（是牛的位置距离）
	c[niu[0].id]++;	// 记录每个编号出现的次数
	while(hh <= tt)
	{
		while(c[q[hh].id] > 1)	// 判断队头，如果队头在队列中重复了，那么将队头弹出，
            					// 并且减少队头编号的数量，保证队列中满足条件并且位置距离最短
		{
			c[q[hh].id]--;
			hh++;
		}
		if(res == nums)	// 当前队列满足条件，比较位置距离
		{
			ress = min(ress, q[tt].x - q[hh].x);
		}
		if(tt == n-1) break;	// 如果最后一头牛已经入队了，并且队头没有牛可以弹出了，循环结束
        // 将新的牛加入到队尾中，并且判断这个牛的编号是不是第一次出现
		q[++tt] = {niu[tt].x, niu[tt].id};
		if(c[niu[tt].id] == 0) res++;
		c[niu[tt].id]++;
	}
	cout << ress << endl;
	
	return 0;
}