Acwing 171. 送礼物(双向dfs + 二分)

题目描述:
达达帮翰翰给女生送礼物,翰翰一共准备了 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]);
	}
}

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Easenyang

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值