求数组所有非空子序列的和

算法思想:
  对每一个数,可以选与不选,那么就可以得到所有的非空子序列的和。

例题:

在这里插入图片描述

class Solution {
    typedef long long LL;
    typedef pair<LL, int> pli;

public:
    long long kSum(vector<int>& nums, int k) {
        int n = nums.size();

        LL sum = 0;
        for(int i = 0; i < n; ++i){
            if(nums[i] > 0) sum += nums[i];
            else nums[i] = -nums[i];
        }

        sort(nums.begin(), nums.end());

        priority_queue<pli> q;
        q.push({sum, 0});

        while(--k){
            auto [sum, i] = q.top();
            q.pop();

            if(i < n){
                q.push({sum - nums[i], i + 1}); //保留 nums[i - 1]
                if(i > 0) q.push({sum + nums[i - 1] - nums[i], i + 1}); //不保留 nums[i - 1]
            }
        }

        return q.top().first;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值