原题链接:https://www.acwing.com/problem/content/173/
题目描述:
达达帮翰翰给女生送礼物,翰翰一共准备了N个礼物,其中第i个礼物的重量是G[i]。
达达的力气很大,他一次可以搬动重量之和不超过W的任意多个物品。
达达希望一次搬掉尽量重的一些物品,请你告诉达达在他的力气范围内一次性能搬动的最大重量是多少。
输入格式
第一行两个整数,分别代表W和N。
以后N行,每行一个正整数表示G[i]。
输出格式
仅一个整数,表示达达在他的力气范围内一次性能搬动的最大重量。
数据范围
1≤N≤46,
1≤W,G[i]≤231−1
输入样例:
20 5
7
5
4
18
1
输出样例:
19
题解
可以注意到这道题的N最大为46, 按照每个物品枚举,每个物品有选和不选两种状态,总共有2^46种状态,肯定会超时。可以分成两步选择,即将物品分成两堆,先将其中一堆所有可能的重量储存起来,在以相同的做法枚举第二堆所有的重量,将两堆重量合并,找到最大值。这里可以先将第一堆排序,然后二分法。
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long LL;
const int N = 1 << 24;
LL w, n;
int k;
LL g[50], weight[N];
LL cnt = 0;
LL ans;
void dfs(int u, int s)
{
if (u == k)
{
weight[cnt ++ ] = s;
return;
}
if ((LL)s + g[u] <= w) dfs(u + 1, (LL)s + g[u]);
dfs(u + 1, s);
}
void dfs2(int u, LL s)
{
if(u == n)
{
int l = 0, r = cnt - 1;
while(l < r)
{
int mid = l + r + 1 >> 1;
if(s + weight[mid] <= w) l = mid;
else r = mid - 1;
}
if (weight[l] + s <= w) ans = max(ans, weight[l] + s);
return;
}
if(s + g[u] <= w) dfs2(u + 1, s + g[u]);
dfs2(u + 1, s);
}
int main()
{
cin >> w >> n;
for (int i = 0; i < n; i ++ ) cin >> g[i];
sort(g, g + n);
reverse(g, g + n);
k = n / 2 + 2;
dfs(0, 0);
sort(weight, weight + cnt);
int t = 1;
for (int i = 1; i < cnt; i ++ )
if (weight[i] != weight[i - 1])
weight[t ++ ] = weight[i];
cnt = t;
dfs2(k, 0);
cout << ans << endl;
return 0;
}