Codeforces 754 D Fedor and coupons【优先队列】

题目：http://codeforces.com/contest/754/problem/D

题意：

从n条线段中取出恰好k条使得交集长度尽可能长，输出最优值和方案。

分析：

感觉这种题做过好多次了，贪心+优先队列就可做。

根据左端点排序，然后依次插入右端点，直到队列中元素有k个，然后当前插入的左端点是最大的，队首的就是右端点最小的。这样维护一下即可。

set和优先队列都可以，set似乎更方便些~~

代码：

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;

struct Coupon {
int l;
int r;
int id;

Coupon(int l, int r, int id): l(l), r(r), id(id) {}

bool operator<(const Coupon& c) {
return l < c.l;
}
};

int main() {
ios_base::sync_with_stdio(0);
cin.tie(nullptr);
int n, k;
cin >> n >> k;
vector<Coupon> e;
for (int i = 0; i < n; i++) {
int l, r;
cin >> l >> r;
e.push_back(Coupon(l, r, i));
}
sort(e.begin(), e.end());
set<pii> q;
int ans = 0;
int when = -1;
for (int i = 0; i < e.size(); i++) {
q.insert(pii(e[i].r, e[i].id));
while (q.size() > k)
q.erase(q.begin());
if (q.size() == k) {
int cur = q.begin()->first - e[i].l + 1;
if (cur > ans) {
ans = cur;
when = i;
}
}
}
cout << ans << endl;
if (ans == 0) {
for (int i = 1; i <= k; i++)
cout << i << " ";
cout << endl;
return 0;
}
q.clear();
for (int i = 0; i < e.size(); i++) {
q.insert(pii(e[i].r, e[i].id));
while (q.size() > k)
q.erase(q.begin());
if (when == i) {
for (auto kv : q)
cout << kv.second + 1 << " ";
cout << endl;
return 0;
}
}
}