Leetcode删除元素后的最小差

Leetcode删除元素后的最小差_第1张图片
Leetcode删除元素后的最小差_第2张图片
思路:
我们先枚举能组成第一部分的最小值,将答案记录在数组中
再枚举能组成第二部分的最大值,将答案记录在数组中,最后在枚举最小值即可,利用优先队列+前缀和实现

class Solution {
public:
    long long minimumDifference(vector<int>& nums) {
        priority_queue<long long>_bg;
        priority_queue<long long , vector<long long>, greater<long long>>_sml;
        vector<long long>sum(nums.size()+1,0);
       
        long long n=nums.size()/3; 
        vector<long long>aser1(n+1,0);
        vector<long long>aser2(n+1,0);
        long long bgs=0,smls=0;
        for(int i=1;i<=nums.size();i++){
            sum[i]=sum[i-1]+nums[i-1];
            if(i<=n){
                _bg.push(nums[i-1]);
            }
            if(i>(2*n)){
                _sml.push(nums[i-1]);
            }
        }
        aser1[0]=sum[n];
        for(int i=n;i<2*n;i++){
            _bg.push(nums[i]);
            bgs+=_bg.top();
            _bg.pop();
            aser1[i-n+1]=sum[i+1]-bgs;
        }
        
        aser2[0]=sum[3*n]-sum[2*n];
        
        for(int i=2*n-1,j=1;i>=n;i--,j++){
            _sml.push(nums[i]);
            smls+=_sml.top();
            _sml.pop();
            aser2[j]=sum[3*n]-sum[i]-smls;
        }
       
        int k=n;
        long long ans=aser1[0]-aser2[k--];
        for(int i=1;i<=n;i++){
            ans=min(ans,aser1[i]-aser2[k--]);
        }
        return ans;
         
    }
};

你可能感兴趣的:(LeetCode刷题,leetcode,算法,职场和发展)