题目描述
思路
- 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;
}