题目描述:
达达帮翰翰给女生送礼物,翰翰一共准备了 N 个礼物,其中第 i 个礼物的重量是 G[i]。
达达的力气很大,他一次可以搬动重量之和不超过 W 的任意多个物品。
达达希望一次搬掉尽量重的一些物品,请你告诉达达在他的力气范围内一次性能搬动的最大重量是多少。
输入格式
第一行两个整数,分别代表 W 和 N。
以后 N 行,每行一个正整数表示 G[i]。
输出格式
仅一个整数,表示达达在他的力气范围内一次性能搬动的最大重量。
数据范围
1 ≤ N ≤ 46,
1 ≤ W, G[i] ≤ 231−1
题目链接:送礼物
分析:
一开始会想到用背包问题的DP思维来解,但是我们注意w若是231−1,即使使用一维数组也会超内存,所以是不合理的。
然后考虑dfs去解决,时间复杂度是O(2n),当 n = 46时,246会达到1013级别,这肯定会超时 ,我们就可以考虑分成两半来dfs,223只会达到106,乘以2,224也才达到107,所有就分两次来dfs,第一次dfs前半部分,第二次dfs后半部分,这样就可以解决了。
思路:
1.先dfs前半部分,将他们能拼成的所有数字放入一个HashSet集合(去重),再新建一个List集合存放Set集合的元素并从小到大排序,以便后面的二分计算。
public static void dfs(int u, long sum) {
if (u>=k) { //k是n的一半
set.add((int)sum);
return;
}
dfs(u+1, sum);
if (sum + g[u] <= w)
dfs(u+1, sum+g[u]);
}
list = new ArrayList<>(set);
Collections.sort(list);
2.dfs后半部分,用后半部分得到的每一个值与前半部分二分找到的最佳值相加,得到结果,再去更新最大结果即可。
public static void dfs2(int u, long sum) {
if (u>=n) {
int l = 0, r = list.size() - 1;
while (l < r) {//二分查找去找前半部分和当前数能拼成的最大值
int mid = l + r + 1 >> 1;
if (list.get(mid) + sum <= w)
l = mid;
else r = mid - 1;
}
res = Math.max(res, (int)(list.get(l)+sum));
return;
}
dfs2(u+1, sum);
if (sum + g[u] <= w)
dfs2(u+1, sum+g[u]);
}
最后再分析一下时间复杂度,dfs的时间复杂度为O(2n),二分的时间复杂度为O(log2n),综合起来就是O(2n·log2n)
完整代码:
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;
import java.util.List;
public class Main {
static int w, n, k;//最大重量、物品数量、一半物品数量
static int[] g;//存放各物品重量
static int res = 0;
static Set<Integer> set = new HashSet<>();//去重的Set集合
static List<Integer> list; //用来排序的List集合
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
w = sc.nextInt(); n = sc.nextInt();
g = new int[n+1];
for (int i = 0; i < n; i++)
g[i] = sc.nextInt();
k = n/2 + 1;
dfs(0, 0);//第一次dfs,找前一半能拼成的所有重量
list = new ArrayList<>(set);//将set集合的元素放入List
Collections.sort(list); //进行排序
dfs2(k, 0); //第二次dfs,找最优结果
System.out.println(res);
}
public static void dfs(int u, long sum) {
if (u>=k) {
set.add((int)sum);
return;
}
dfs(u+1, sum);
if (sum + g[u] <= w)
dfs(u+1, sum+g[u]);
}
public static void dfs2(int u, long sum) {
if (u>=n) {
int l = 0, r = list.size() - 1;
while (l < r) {
int mid = l + r + 1>> 1;
if (list.get(mid) + sum <= w)
l = mid;
else r = mid-1;
}
res = Math.max(res, (int)(list.get(l)+sum));//更新结果
return;
}
dfs2(u+1, sum);
if (sum + g[u] <= w)
dfs2(u+1, sum+g[u]);
}
}