本文参考了hh_is_vegetable大佬的文章:文章链接
描述
一个数组中有若干正整数,将此数组划分为两个子数组,使得两个子数组各元素之和a,b的差最小,对于非法输入应该输出ERROR。
输入描述:
数组中的元素
输出描述:
降序输出两个子数组的元素和
示例
Input
10 20 30 10 10
10 20 abc 10 10
Output
40 40
ERROR
思路:
也就是求数组中哪些元素的和,能够在小于等于sum/2的条件下最大。于是,看似是01背包的题。
但是,01背包的体积V是有限制的,这里V可以很大!int范围内都可以取到,也就是需要232B=4GB的空间,不可能dp。
所以考虑DFS,但是DFS的复杂度又是O(2n),必须要剪枝。
下面第13行的剪枝思路真是妙到不行:先对数组从大到小排序,那么DFS的过程中,到了第i
个元素选不选取的子问题。当前子问题的和记为val
,最大和记为max_val
,用于储存数据的数组为vec
,如果val+(vec.size()-i)*vec[i]<=max_val
,那么,即使选取后面的所有元素,都不可能更新max_val
了,所以可以直接return,非常漂亮。
另外还有一些零碎的剪枝,比如到了max_val==half
的时候就可以一路return回去(其实,注释掉13行的剪枝和30行的排序,只用这个剪枝都能88ms左右AC,上面那个技巧性很强的剪枝可以直接压到19ms)。
这道题的IO处理也值得学习,stringstream和sscanf、sprintf一样都是处理字符串的好工具。
#include<bits/stdc++.h>
using namespace std;
int sum,half;
int max_half=0;
void DFS(vector<int>