Codeforces 1203F (贪心, DP)

题意:有n个任务,你的初始rating是m, 这n个任务有两个指标:完成这项任务所需的最低rating(a[i]),以及完成这项任务后rating的变化(可能为负)(b[i])。rating不能为负。F1:问是否存在一种任务完成顺序,始得所有任务都可以被完成,。F2:你可以任意选择一些任务去完成,问最多可以完成多少任务。

思路:

F1: 首先任务分成两部分:涨rating的和降rating的。对于涨rating的任务,直接按照a[i]从小到大排序即可。为什么呢?因为完成rating之后rating一定不会比原来小,即原来能完成的任务在完成其它任务后也一定能完成,如果可以全部完成,就开始考虑降rating的部分。我们考虑一下降rating的部分,我们考虑每一个任务执行之后rating的下界,然后按下界从大到小排序,即优先选择下界大的任务,下界越高对后面的影响越小。

代码:

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 110;
struct node {
	int x, y;
};
vector<node> a, b;
bool cmp1(node t1, node t2) {
	return t1.x < t2.x;
}
bool cmp2(node t1, node t2) {
	return (max(t1.x, -t1.y) + t1.y) > (max(t2.x, -t2.y) + t2.y);
}
int main() {
	int n, m, cnt1 = 0, cnt2 = 0;
	scanf("%d%d", &n, &m);
	int sum = 0;
	node tmp;
	for (int i = 1; i <= n; i++) {
		scanf("%d%d", &tmp.x, &tmp.y);
		if(tmp.y >= 0) a.push_back(tmp);
		else b.push_back(tmp);
	}
	sort(a.begin(), a.end(), cmp1);
	sort(b.begin(), b.end(), cmp2);
	int ans = 0;
	for (int i = 0; i < a.size(); i++) {
		if(m < a[i].x) ans = -1;
		m += a[i].y;
	}
	for (int i = 0; i < b.size(); i++) {
		if(m < b[i].x) ans = -1;
		m += b[i].y;
	}
	if(ans == -1) printf("NO\n");
	else printf("YES\n");
} 

F2:很明显这是一个背包DP,但是直接DP是O(n * n * m)的,因为每一个任务的选择会影响后面的任务。这样DP会超时。通过F1的结论我们得知,如果我们把降rating的部分按照下界排序,优先选择上界高的任务,这样我们DP转移时只需枚举当前任务选择还是不选择就行了,因为这种决策是最优的,所以不需再考虑之前完成的任务的影响,这样O(n * m)的复杂度便可完成DP过程。

代码:

#include <bits/stdc++.h>
#define pii pair<int, int>
using namespace std;
const int maxn = 600010;
int dp[110][maxn];
bool cmp(pii t1, pii t2) {
	return (t1.first + t1.second) > (t2.first + t2.second);
}
vector<pii> a, b;
int main() {
	int n, m;
	scanf("%d%d", &n, &m);
	pii tmp;
	for (int i = 1; i <= n; i++) {
		scanf("%d%d", &tmp.first, &tmp.second);
		if(tmp.second >= 0) a.push_back(tmp);
		else b.push_back(tmp);
	}
	sort(a.begin(), a.end());
	sort(b.begin(), b.end(), cmp);
	int mx = m, tot = 0;
	for (int i = 0; i < a.size(); i++) {
		 if(a[i].first <= mx) {
		 	mx += a[i].second;
		 	tot++;
		 }
	}
	int ans = 0;
	dp[0][mx] = tot;
	for (int i = 0; i < b.size(); i++) {
		for (int j = 0; j <= mx; j++) {
			if(j >= b[i].first && j + b[i].second >= 0) {
				dp[i + 1][j + b[i].second] = max(dp[i + 1][j + b[i].second], dp[i][j] + 1);
			}
			dp[i + 1][j] = max(dp[i + 1][j], dp[i][j]);
		}
	}
	for (int i = 0; i <= mx; i++) {
		ans = max(ans, dp[b.size()][i]);
	}
	printf("%d\n", ans);
} 

  

猜你喜欢

转载自www.cnblogs.com/pkgunboat/p/11361313.html