思路
因为数据范围为40,因此需要用到双向dfs,如果数据范围小于20时就可以直接用dfs进行爆搜。
思路是将整个数组分为两段,前面一段用dfs爆搜出全部组合的和并存储到数组中。然后将数组排序,再对后面一段进行爆搜,然后当搜索完最后一个元素后就进行二分查找前面与目标值的差最小的绝对值。
问题描述
给你一个整数数组 nums 和一个目标值 goal 。
你需要从 nums 中选出一个子序列,使子序列元素总和最接近 goal 。也就是说,如果子序列元素和为 sum ,你需要 最小化绝对差 abs(sum - goal) 。
返回 abs(sum - goal) 可能的 最小值 。
注意,数组的子序列是通过移除原始数组中的某些元素(可能全部或无)而形成的数组。
代码
const int N = 200000010;
int q[N];
class Solution {
public:
int ans = 1e9, cnt = 0;
int goal;
int n;
void dfs1(vector<int>& nums, int u, int s){
if(u == (n+1)/2){
q[cnt++] = s;
return;
}
dfs1(nums, u + 1, s);
dfs1(nums, u + 1, s + nums[u]);
}
void dfs2(vector<int>& nums, int u, int s){
if(u == n){
int l = 0, r = cnt - 1;
while(l < r){
int mid = l + r + 1 >> 1;
if(q[mid] + s <= goal) l = mid;
else r = mid - 1;
}
ans = min(ans, abs(q[l] + s - goal));
if(l + 1 < cnt){
ans = min(ans, abs(q[l+1] + s - goal));
}
return;
}
dfs2(nums, u + 1, s);
dfs2(nums, u + 1, s + nums[u]);
}
int minAbsDifference(vector<int>& nums, int _goal) {
goal = _goal;
n = nums.size();
dfs1(nums, 0, 0);
sort(q, q+cnt);
dfs2(nums, (n+1)/2, 0);
return ans;
}
};