CF 1249D1 - Too Many Segments (easy version) 贪心+排序+set的使用

#include <bits/stdc++.h>

using namespace std;
const int maxn = 2e5+10;

struct Node{
    int r;
    int idx;
};
bool operator < (Node a,Node b){
        if(a.r != b.r)
            return a.r<b.r;
        return a.idx<b.idx;
}

vector<Node> segs[maxn];
set<int> ans;
int n,k,x,y;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>k;
    for(int i=0; i<n; i++){
        cin>>x>>y;
        Node temp;
        temp.r = y;
        temp.idx = i;
        segs[x].push_back(temp);
    }
    set<Node> s;
    for(int i=0; i<maxn; i++){
        while(s.size() && (*s.begin()).r<i ){
            s.erase(s.begin());
        }
        for(Node it: segs[i]){
            s.insert(it);
        }
        while(s.size() > k){
            ans.insert( (*s.rbegin()).idx );
            s.erase( *(s.rbegin()) );
        }
    }
    cout<<ans.size()<<endl;
    for(int it : ans){
        cout<<it+1<<" ";
    }
    cout<<endl; 
}


//c.begin() 返回一个迭代器，它指向容器c的第一个元素

// c.end() 返回一个迭代器，它指向容器c的最后一个元素的下一个位置

// c.rbegin() 返回一个逆序迭代器，它指向容器c的最后一个元素

// c.rend() 返回一个逆序迭代器，它指向容器c的第一个元素前面的位置