https://www.acwing.com/problem/content/113/
有 NN 头牛在畜栏中吃草。
每个畜栏在同一时间段只能提供给一头牛吃草,所以可能会需要多个畜栏。
给定 NN 头牛和每头牛开始吃草的时间 AA 以及结束吃草的时间 BB,每头牛在 [A,B][A,B] 这一时间段内都会一直吃草。
当两头牛的吃草区间存在交集时(包括端点),这两头牛不能被安排在同一个畜栏吃草。
求需要的最小畜栏数目和每头牛对应的畜栏方案。
输入格式
第 11 行:输入一个整数 NN。
第 2..N+12..N+1 行:第 i+1i+1 行输入第 ii 头牛的开始吃草时间 AA 以及结束吃草时间 BB,数之间用空格隔开。
输出格式
第 11 行:输入一个整数,代表所需最小畜栏数。
第 2..N+12..N+1 行:第 i+1i+1 行输入第 ii 头牛被安排到的畜栏编号,编号是从 11 开始的 连续 整数,只要方案合法即可。
数据范围
1≤N≤500001≤N≤50000,
1≤A,B≤10000001≤A,B≤1000000
输入样例:
5
1 10
2 4
3 6
5 8
4 7
输出样例:
4
1
2
3
2
4
这题有点复杂,就是数据之间的练习,最初我想到用结构体,排序再循环判断,但是超时了,看了题解,了解了优先队列是可以自动排序列的,这样在对每头牛吃草的时间段进行升序排序后,就可以每头牛只和队头元素进行比较就可以了,这样时间复杂度就是n,顺利ac,
优先队列:
和队列基本操作相同:
- top 访问队头元素
- empty 队列是否为空
- size 返回队列内元素个数
- push 插入元素到队尾 (并排序)
- emplace 原地构造一个元素并插入队列
- pop 弹出队头元素
- swap 交换内容
//升序队列
priority_queue <int,vector<int>,greater<int> > q;
//降序队列
priority_queue <int,vector<int>,less<int> >q;
//greater和less是std实现的两个仿函数(就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator(),这个类就有了类似函数的行为,就是一个仿函数类了)
//对于基础类型 默认是大顶堆
priority_queue<int> a;
//等同于 priority_queue<int, vector<int>, less<int> > a;
ac代码:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
using namespace std;
int n;
typedef pair<int,int> PII;
pair<PII,int> cows[50010];//记录每头牛吃草的时间段,奶牛编号
int id[50010];//记录每头牛的栅栏号;
int main(){
cin>>n;
for(int i=0;i<n;i++){
cin>>cows[i].first.first>>cows[i].first.second;
cows[i].second=i;
}
sort(cows,cows+n);//保证按每头牛开始吃草时间升序排列
priority_queue<PII,vector<PII>,greater<PII> > qu;//{吃草结束的时间,畜栏编号}
for(int i=0;i<n;i++){
if(qu.empty()||cows[i].first.first<=qu.top().first){
id[cows[i].second]=qu.size()+1;
PII a1;
a1.first=cows[i].first.second;
a1.second=qu.size()+1;
qu.push(a1);
}
else{
PII a=qu.top();
qu.pop();
PII b;
b.first=cows[i].first.second;
b.second=a.second;
id[cows[i].second]=a.second;
qu.push(b);
}
}
cout<<qu.size()<<endl;
for(int i=0;i<n;i++){
cout<<id[i]<<endl;
}
return 0;
}