P5194 [USACO05DEC] Scales S

[USACO05DEC] Scales S

题目描述

约翰有一架用来称牛的体重的天平。与之配套的是 $N(1\le N\le 1000)$ 个已知质量的砝码（所有砝码质量的数值都在 $32$ 位带符号整数范围内）。

每次称牛时，他都把某头奶牛安置在天平的某一边，然后往天平另一边加砝码，直到天平平衡，于是此时砝码的总质量就是牛的质量（约翰不能把砝码放到奶牛的那边，因为奶牛不喜欢称体重，每当约翰把砝码放到她的蹄子底下，她就会尝试把砝码踢到约翰脸上）。

天平能承受的物体的质量不是无限的，当天平某一边物体的质量大于 $C(1\le C\le 2^{30})$ 时，天平就会被损坏。砝码按照它们质量的大小被排成一行。并且，这一行中从第 $3$ 个砝码开始，每个砝码的质量至少等于前面两个砝码（也就是质量比它小的砝码中质量最大的两个）的质量的和。

约翰想知道，用他所拥有的这些砝码以及这架天平，能称出的质量最大是多少。由于天平的最大承重能力为 $C$，他不能把所有砝码都放到天平上。

现在约翰告诉你每个砝码的质量，以及天平能承受的最大质量，你的任务是选出一些砝码，使它们的质量和在不压坏天平的前提下是所有组合中最大的。

输入格式

第 $1$ 行输入两个用空格隔开的正整数 $N$ 和 $C$。

第 $2$ 到 $N+1$ 行：每一行仅包含一个正整数，即某个砝码的质量。保证这些砝码的质量是一个不下降序列。

输出格式

输出一个正整数，表示用所给的砝码能称出的不压坏天平的最大质量。

样例

样例输入

3 15
1
10
20

样例输出

11

开始思路

本题为dfs题，初看很简单，dfs的参数为当前的总和$sum$和现在的下标$i$，判断条件为当前$sum$是否大于$C$

代码（70分）

#include <bits/stdc++.h>
#define endl '\n'
#define int long long
using namespace std;
int n, a[1010], c, mx;
void dfs(int sum, int tmp) {
	if (sum > c)	return;
	mx = max(mx, sum);
	for (int i = tmp; i >= 1; i--) {
		dfs(sum + a[i], i - 1);
	}
}
signed main() {
	ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
	cin >> n >> c;
	for (int i = 1; i <= n; i++)
		cin >> a[i];
	dfs(0, n);
	cout << mx;
	return 0;
}

结果是有三个点TLE超时了

反思

本题照模板去套dfs不能过全部点，需要一点以空间换时间的操作，即记忆化+剪枝
我们可以开一个sum[ ]数组，起到前缀和的作用，根据题意：这一行中从第 3 个砝码开始，每个砝码的质量至少等于前面两个砝码（也就是质量比它小的砝码中质量最大的两个）的质量的和，如果当前tmp前面所有的元素加起来都没有当前的mx大，剪枝即可。

AC代码

#include <bits/stdc++.h>
#define endl '\n'
#define int long long
using namespace std;
int n, a[1010], c, mx, sm[1010];
void dfs(int sum, int tmp) {
	if (sum > c)	return;
	if (sum + sm[tmp] < mx)	return;	//剪枝，如果前面的所有和还没有现在的mx大，那就不用比了
	mx = max(mx, sum);
	for (int i = tmp; i >= 1; i--) {
		dfs(sum + a[i], i - 1);
	}
}
signed main() {
	ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
	cin >> n >> c;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		sm[i] = sm[i - 1] + a[i];
	}
	dfs(0, n);
	cout << mx;
	return 0;
}

附上思考笔记