首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >LeetCode 2035. 将数组分成两个数组并最小化数组和的差(状态压缩DP)

LeetCode 2035. 将数组分成两个数组并最小化数组和的差(状态压缩DP)

作者头像
Michael阿明
发布2022-01-07 11:13:21
发布2022-01-07 11:13:21
2.7K00
代码可运行
举报
运行总次数:0
代码可运行

文章目录

1. 题目

给你一个长度为 2 * n 的整数数组。 你需要将 nums 分成 两个 长度为 n 的数组,分别求出两个数组的和,并 最小化 两个数组和之 差的绝对值 。 nums 中每个元素都需要放入两个数组之一。

请你返回 最小数组和之差

代码语言:javascript
代码运行次数:0
运行
复制
示例 1:
输入:nums = [3,9,7,3]
输出:2
解释:最优分组方案是分成 [3,9] 和 [7,3] 。
数组和之差的绝对值为 abs((3 + 9) - (7 + 3)) = 2 。

示例 2:
输入:nums = [-36,36]
输出:72
解释:最优分组方案是分成 [-36] 和 [36] 。
数组和之差的绝对值为 abs((-36) - (36)) = 72 。

示例 3:
输入:nums = [2,-1,0,4,-2,-9]
输出:0
解释:最优分组方案是分成 [2,4,-9] 和 [-1,0,-2] 。
数组和之差的绝对值为 abs((2 + 4 + -9) - (-1 + 0 + -2)) = 0 。
 
提示:
1 <= n <= 15
nums.length == 2 * n
-10^7 <= nums[i] <= 10^7

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/partition-array-into-two-arrays-to-minimize-sum-difference 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • 数组折半,分别对一半进行状态枚举
  • 枚举一边取的数的个数,将左右的满足二进制位个数的状态取出,排序,双指针求解最接近的
  • 时间复杂度 O(n*2^n)
代码语言:javascript
代码运行次数:0
运行
复制
class Solution {
public:
    int minimumDifference(vector<int>& nums) {
        int n = nums.size()/2;
        int sum = accumulate(nums.begin(), nums.end(), 0);
        vector<int> a(nums.begin(),nums.begin()+n), b(nums.begin()+n,nums.end());
        vector<int> stateSum1 = getsum(a);//获取所有状态的和
        vector<int> stateSum2 = getsum(b);
        int dis = INT_MAX;
        for(int x = 0; x <= n; ++x)
        { // 第一个数组取x个数
            int y = n-x; // 第二个数组取y个数
            vector<int> s1, s2;//把两边取出来的和存储
            for(int state = 0; state < (1<<n); ++state)
            {
                if(count1(state)==x)
                    s1.push_back(stateSum1[state]);
            }
            for(int state = 0; state < (1<<n); ++state)
            {
                if(count1(state)==y)
                    s2.push_back(stateSum2[state]);
            }
            sort(s1.begin(), s1.end());//排序,双指针,求解最接近的
            sort(s2.begin(), s2.end());
            int len1 = s1.size(), len2 = s2.size();
            int i = 0, j = len2-1;
            while(i < len1 && j >= 0)
            {
                dis = min(dis, abs(sum-2*(s1[i]+s2[j])));
                if(s1[i]+s2[j] > sum-(s1[i]+s2[j]))
                    j--;
                else if(s1[i]+s2[j] < sum-(s1[i]+s2[j]))
                    i++;
                else
                    return 0;
            }
        }
        return dis;
    }
    vector<int> getsum(vector<int>& num)
    {
        int n = num.size();
        vector<int> ans(1<<n);
        for(int i = 0; i < n; ++i)
        {
            ans[1<<i] = num[i]; // 初始化
        }
        for(int state = 1; state < (1<<n); ++state)
        {
            int prevstate = state&(state-1);
            ans[state] = ans[state-prevstate] + ans[prevstate];
        }
        return ans;
    }
    int count1(int x)
    { // 计算二进制1的个数
        int ans = 0;
        while(x)
        {
            ans++;
            x = x&(x-1);
        }
        return ans;
    }
};

692 ms 76.4 MB C++


我的CSDN博客地址 https://michael.blog.csdn.net/

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021/10/10 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 文章目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档