思路:
我们先枚举能组成第一部分的最小值,将答案记录在数组中
再枚举能组成第二部分的最大值,将答案记录在数组中,最后在枚举最小值即可,利用优先队列+前缀和实现
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;
}
};