CF1471-C. Strange Birthday Party

CF1471-C. Strange Birthday Party

题意：

你要举办一场生日派对。派对有\(n\)个人，每个人都有一个数字\(k_i\)。超市有\(m\)件礼物，购买每件礼物需要花费\(\$c_j(c_1<c_2<...<c_m)\)，且每个礼物只有一件。你要给这\(n\)个人发东西，对于第\(i\)个人，你有两种选择：

1. 给第\(i\)个人发礼物，那么给这个人的礼物要求\(j<=k_i\)，即礼物的编号不能超过这个人的数字\(k_i\);

2. 直接给这个人\(\$c_{k_i}\).

现在要你求出最小的花费。

---

思路：

按花费从小到大依次分配礼物。假设现在要分配的礼物的编号是\(cur\)，利用贪心的思想，当\(cur<=k_i\)时，依次将花费最小的礼物分给\(k_i\)最大的人；当\(cur>k_i\)时，直接给这位朋友\(\$c_{k_i}\)。

原因如下：当\(cur<=k_i\)时，有\(c_{cur}<=c_{k_i}\)；当\(cur>k_i\)时，有\(c_{k_i}<c_{cur}\)，每次都取两者中较小的。

---

AC代码：

#include <cstdio>
#include <cstring>
#include <algorithm>

typedef long long ll;

const int Maxn = 300005;

int a[Maxn], b[Maxn];

void solve() {
	int n, m;
	scanf("%d %d", &n, &m);
	for (int i = 1; i <= n; i++) {
		scanf("%d", a + i);
	}
	for (int i = 1; i <= m; i++) {
		scanf("%d", b + i);
	}
	std::sort(a + 1, a + n + 1);
	ll ans = 0;
	int cur = 1;
	for (int i = n; i > 0; i--) {
		if (a[i] > cur) {
			ans += b[cur++];
		} else {
			ans += b[a[i]];
		}
	}
	printf("%lld\n", ans);
}

int main() {
	int T;
	scanf("%d", &T);
	while (T--) {
		solve();
	}

	return 0;
}