题目:http://codeforces.com/contest/1070/problem/C
很有意思的题…
在1到n天里,每天需求k块cpu,在m种有时间限制的供应方案里挑选,尽量满足需求的前提下花费最小。
显然,对于每一天,都需要求出一个最优解。而最优解的求得,容易想到是一个简单的贪心,优先选择便宜的,直到满足需求。
但是这里要求得每天的结果,并不是一次排序可以搞定的,那么维护这种序列需要动态插入和删除,并且保持有序…线段树可以做到。
线段树维护一个每种方案的花费区间,然后在插入的时候插入到对应的位置,就完成了排序。
查询的时候找到第k个的位置,并且加上前k - 1个的花销,就求出了一天的解。
ac代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 5;
vector<int> st[maxn], ed[maxn];
struct supply {
//记录l,r方便debug
int l, r;
ll c, p;
} sp[200005];
ll val[maxn << 2], sz[maxn << 2];
void pushUp(int i) {
val[i] = val[i << 1] + val[i << 1 | 1];
sz[i] = sz[i << 1] + sz[i << 1 | 1];
}
void update(int i, int l, int r, ll pos, ll v) {
if(l == r) {
val[i] += v * pos;
sz[i] += v;
return;
}
int mid = (l + r) >> 1;
if(pos <= mid) {
update(i << 1, l, mid, pos, v);
} else {
update(i << 1 | 1, mid + 1, r, pos, v);
}
pushUp(i);
}
ll query(int i, int l, int r, int k) {
if(l == r) {
return min(1LL * k, sz[i]) * l;
}
int mid = (l + r) >> 1;
if(k <= sz[i << 1]) {
return query(i << 1, l, mid, k);
}
return val[i << 1] + query(i << 1 | 1, mid + 1, r, k - sz[i << 1]);
}
int main() {
int n, k, m;
scanf("%d%d%d", &n, &k, &m);
for(int i = 1; i <= m; i++) {
scanf("%d%d%d%d", &sp[i].l, &sp[i].r, &sp[i].c, &sp[i].p);
st[sp[i].l].push_back(i);
ed[sp[i].r].push_back(i);
}
ll ans = 0;
for(int i = 1; i <= n; i++) {
// 开始的区间加入
for(int j = 0; j < st[i].size(); j++) {
int e = st[i][j];
update(1, 1, maxn, sp[e].p, sp[e].c);
}
ans += query(1, 1, maxn, k);
// 结束的区间删除
for(int j = 0; j < ed[i].size(); j++) {
int e = ed[i][j];
update(1, 1, maxn, sp[e].p, -sp[e].c);
}
}
printf("%lld\n", ans);
return 0;
}