NEERC C. Cloud Computing（线段树）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Cymbals/article/details/83304720

题目：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;
}