其他分享
首页 > 其他分享> > 状压DP-1755. 最接近目标值的子序列和

状压DP-1755. 最接近目标值的子序列和

作者:互联网

问题描述

给你一个整数数组 nums 和一个目标值 goal 。
你需要从 nums 中选出一个子序列,使子序列元素总和最接近 goal 。也就是说,如果子序列元素和为 sum ,你需要 最小化绝对差 abs(sum - goal) 。
返回 abs(sum - goal) 可能的 最小值 。
注意,数组的子序列是通过移除原始数组中的某些元素(可能全部或无)而形成的数组。

示例 1:
输入:nums = [5,-7,3,5], goal = 6
输出:0
解释:选择整个数组作为选出的子序列,元素和为 6 。
子序列和与目标值相等,所以绝对差为 0 。
示例 2:
输入:nums = [7,-9,15,-2], goal = -5
输出:1
解释:选出子序列 [7,-9,-2] ,元素和为 -4 。
绝对差为 abs(-4 - (-5)) = abs(1) = 1 ,是可能的最小值。
示例 3:
输入:nums = [1,2,3], goal = -7
输出:7
 
提示:
1 <= nums.length <= 40
-107 <= nums[i] <= 107
-109 <= goal <= 109

问题求解

规模是40 --> 折半状压DP即可。

class Solution:
    def minAbsDifference(self, nums: List[int], goal: int) -> int:
        n = len(nums)
        lnums = nums[:n//2]
        rnums = nums[n//2:]
        lsum = [0] * (1 << len(lnums))
        rsum = [0] * (1 << len(rnums))

        for i in range(1, len(lsum)):
            for j in range(len(lnums)):
                if i & (1 << j) != 0:
                    lsum[i] = lsum[i ^ (1 << j)] + lnums[j]
                    break
        
        for i in range(1, len(rsum)):
            for j in range(len(rnums)):
                if i & (1 << j) != 0:
                    rsum[i] = rsum[i ^ (1 << j)] + rnums[j]
                    break
        
        res = float("inf")

        for i in range(len(lsum)):
            res = min(res, abs(lsum[i] - goal))
        for i in range(len(rsum)):
            res = min(res, abs(rsum[i] - goal))
        
        lsum.sort()
        rsum.sort()
        i = 0
        j = len(rsum) - 1
        while i < len(lsum) and j >= 0:
            res = min(res, abs(lsum[i] + rsum[j] - goal))
            if lsum[i] + rsum[j] >= goal:
                j -= 1
            else:
                i += 1
        
        return res

标签:goal,nums,状压,示例,abs,数组,序列,1755,DP
来源: https://www.cnblogs.com/hyserendipity/p/16607195.html