Codeforces 1249D2 Too Many Segments (hard version)

思路：

1.用数组存储以每个点为起点的线段，线段由线段的编号(first)和它最右边可以到达的点(second)组成；
2.自定义排序规则，右侧较远的线段排在前面，同时右侧一样远的按编号排序（否则set会认为重复，用multiset也不行，erase时会一次移除多个）；
3.依次从左往右遍历，采用贪心算法，用set动态存储当前时刻覆盖该点的线段，从rbegin开始，如果不能达到该点，依次移除；同时该点若是坏点，从begin开始移除，直到该点不是坏点为止；

代码：

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> p;
#define m_p(a,b) make_pair(a,b)
#define p_b(a) push_back(a)
#define rp(i,n) for(int i=0;i<n;i++)
#define rpn(i,n) for(int i=1;i<=n;i++)
const int MAX_N=200005;//
vector<p> pot[MAX_N];//以该点为起点的线段 first是线段编号 second是线段右侧可达的最远点
struct Rule{
	bool operator()(const p &a,const p &b)const{
		return a.second==b.second?a.first<b.first:a.second>b.second;
	}//右侧远的排在前面 
};
set<p,Rule> st;
int main(){
	int n,k;
	cin>>n>>k;
	rpn(i,n){
		int l,r;
		cin>>l>>r;
		pot[l].p_b(m_p(i,r));
	}
	vector<int> v;//用于输出 
	rp(i,MAX_N){//最右侧到不了i的线段被移除 
		while(st.size()&&(*st.rbegin()).second<i) st.erase(*st.rbegin());
		for(auto e:pot[i]) st.insert(e);
		while(st.size()>k){
			v.p_b((*st.begin()).first);
			st.erase(st.begin());
		}
	}
	cout<<v.size()<<'
';
	if(v.size()) cout<<v[0];
	for(int i=1;i<v.size();i++) cout<<' '<<v[i]; 
	return 0;
}