[USACO05DEC] Scales S
题目描述
约翰有一架用来称牛的体重的天平。与之配套的是 $ N \ ( 1 \leq N \leq 1000 ) $ 个已知质量的砝码(所有砝码质量的数值都在 32 32 32 位带符号整数范围内)。
每次称牛时,他都把某头奶牛安置在天平的某一边,然后往天平另一边加砝码,直到天平平衡,于是此时砝码的总质量就是牛的质量(约翰不能把砝码放到奶牛的那边,因为奶牛不喜欢称体重,每当约翰把砝码放到她的蹄子底下,她就会尝试把砝码踢到约翰脸上)。
天平能承受的物体的质量不是无限的,当天平某一边物体的质量大于 $ C \ ( 1 \leq C \leq 2^{30} ) $ 时,天平就会被损坏。砝码按照它们质量的大小被排成一行。并且,这一行中从第 3 3 3 个砝码开始,每个砝码的质量至少等于前面两个砝码(也就是质量比它小的砝码中质量最大的两个)的质量的和。
约翰想知道,用他所拥有的这些砝码以及这架天平,能称出的质量最大是多少。由于天平的最大承重能力为 C C C,他不能把所有砝码都放到天平上。
现在约翰告诉你每个砝码的质量,以及天平能承受的最大质量,你的任务是选出一些砝码,使它们的质量和在不压坏天平的前提下是所有组合中最大的。
输入格式
第 1 1 1 行输入两个用空格隔开的正整数 $ N $ 和 $ C $。
第 2 2 2 到 $ N+1 $ 行:每一行仅包含一个正整数,即某个砝码的质量。保证这些砝码的质量是一个不下降序列。
输出格式
输出一个正整数,表示用所给的砝码能称出的不压坏天平的最大质量。
样例 #1
样例输入 #1
3 15
1
10
20
样例输出 #1
11
代码
#include<bits/stdc++.h>
using namespace std;
const int M=1e7;
long long a[10000],n,c,sum[M],maxn=-1e8;
void dfs(long long sumn,long long st){
maxn = max(maxn,sumn);
for (int i=st;i>=1;i--){
if (sumn+a[i]>c) continue;
if (sumn+sum[i]<=c and i==st){
maxn = max(maxn,sumn+sum[i]);
return;
}
dfs(sumn+a[i],i-1);
}
return;
}
int main(){
cin>>n>>c;
for (int i=1;i<=n;i++) {
cin>>a[i];
sum[i]=sum[i-1]+a[i];
}
dfs(0,n);
cout<<maxn;
return 0;
}
分析
这段代码使用了深度有限搜索(DFS)的方法来解决“选数问题”,其主要思想是枚举所有可能的子集,然后递归搜索得到结果。对于每个状态,我们需要记录当前已选取数的和 sumn
和上一个选取的数的下标 st
,然后在当前位置 i
枚举所有可能的候选数,更新 sumn
的值并继续往下递归搜索,直到结束。
以下是具体的实现过程:
-
首先定义一个全局变量
maxn
,表示最大可选数的和。 -
读入数据时,计算序列前缀和
sum[i] = sum[i-1] + a[i]
,使后续搜索速度加快。 -
定义 DFS 函数
dfs(long long sumn,long long st)
,表示已经选取的数字之和为sumn
,当前处理到的位置为st
。 -
对于当前的状态,首先用
maxn
更新其最大值。 -
设定
for
循环,从后往前遍历候选数。 -
做出两个判断。
- 如果当前候选数大于可选数的范围,则跳过,
sumn+a[i]>c
; - 如果在剩下的所有可选数字中的和不足以超过
maxn
,则可以舍弃这些数字直接返回。剪枝
-
对于仍在可选范围内的候选数,更新状态并继续往下递归搜索。
-
在搜索结束后,输出
maxn
的值即可。
部分由chatgpt生成