Bootstrap

P5194 [USACO05DEC]Scales S

题目描述

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

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

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

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

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

输入格式

第 11 行输入两个用空格隔开的正整数 NN 和 CC。

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

输出格式

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

输入输出样例

输入 #1

3 15
1
10
20

输出 #1

11

本题可以直接用 0 - 1背包解决,dp[j] = max(dp[j],dp[j-c[i]] + w[i]);

在这个我介绍的时DFS:

刚开始直接用dfs暴力,这个数有1000,分支过多会超时

//N个质量的砝码 
//大于C时有损坏
//第3个大于前2个之和
#include<iostream>
using namespace std;
const int N = 1100;
int n,ans = 0,sum = 0;
int a[N], C,visited[N];
void dfs() {
	if (sum > C)
	{
		return;
	}
	else {
		ans = max(ans, sum);
		for (int j = 1; j <= n; ++j) {
			if (!visited[j]) {
				sum += a[j];
				visited[j]++;
				dfs();
				sum -= a[j];
				visited[j]--;
			}
		}
	}
}
int main() {
	cin >> n >> C;
	for (int i = 1; i <= n; i++) cin >> a[i];
	dfs();
	cout << ans;
	return 0;
}

这时想到有没有什么优化方法

题意中有这么一句话:第 3 个砝码开始,每个砝码的质量至少等于前面两个砝码(也就是质量比它小的砝码中质量最大的两个)的质量的和。
 取一段区间的和最大即可 。这时想到了前缀和 优化 dfs

#include<iostream>
#include<algorithm>
using namespace std;
long long sum[1005], a[1005], ans, n, c;
void dfs(int cur, int x) {
	if (ans < x) {
		ans = x;
	}
	if (sum[cur] + x <= c)
	{
		ans = max(ans, sum[cur] + x);
		return;
	}
	//每个砝码的质量至少等于前面两个砝码的质量的和
	
	
	if (x + a[cur] <= c) {
		dfs(cur - 1, x + a[cur]); //选择这个数
	}
	dfs(cur - 1, x);//不选这个数
}
int main() {
	cin >> n >> c;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		sum[i] = sum[i - 1] + a[i];
	}
	dfs(n, 0);
	cout << ans << endl;
	return 0;
}

;