1. 程式人生 > 其它 >演算法:leecode5675.最接近目標值的子序列和(雙向dfs)

演算法:leecode5675.最接近目標值的子序列和(雙向dfs)

技術標籤:演算法資料結構演算法c++雙向dfsleecode5675

思路

因為資料範圍為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; } };

原題連結