一、题目
1、题目描述
给你一个下标从 0 开始的整数数组 nums ,它包含 3 * n 个元素。可以从 nums 中删除 恰好 n 个元素,剩下的 2 * n 个元素将会被分成两个 相同大小 的部分。
前面 n 个元素属于第一部分,它们的和记为 sumfirst。
后面 n 个元素属于第二部分,它们的和记为 sumsecond。
两部分和的 差值 记为 sumfirst - sumsecond。请你返回删除 n 个元素之后,剩下两部分和的 差值的最小值 是多少。
样例输入:[7,9,5,8,1,3]
样例输出:1
2、基础框架
- C++ 版本给出的基础框架代码如下:
class Solution {
public:
long long minimumDifference(vector<int>& nums) {
}
};
3、原题链接
二、解题报告
1、思路分析
(
1
)
(1)
(1) kua ! 数组切成两半!左边一半取 n 个,右边一半取 n 个。
(
2
)
(2)
(2) kua 这一刀总共有 n+1 种 取法,可以从第 n 元素前面 kua 一刀!也可以从 2n 个元素前面 kua 一刀(下标从0开始)。
(
3
)
(3)
(3) 我们观察任何的一刀下去,发生了什么???
(
4
)
(4)
(4) 这一刀给下去之后,问题就明朗了,我们只需要在这一刀的左边取 n 个最小的值,在这一刀的右边取 n 个最大的值。
(
5
)
(5)
(5) 于是问题转变成了求给定一个 2*n 个元素的数组,求前 i 个元素的最小 n 元素和。滑动窗口 + 优先队列 就可以实现了。
(
5.1
)
(5.1)
(5.1) 首先对于前 n 全部塞入优先队列(大顶堆),并且对它们进行和的累加,存放在 first[n-1]
中,实现如下:
first[n-1] = 0;
for(i = 0; i < n; ++i) {
first[n-1] += nums[i];
maxq.push( nums[i] );
}
(
5.2
)
(5.2)
(5.2) 不断向右滑动窗口,每次加入一个数nums[i]
,并且放入堆中,然后从堆中弹出一个最大值,并且将它从 first[i]
中减去。实现如下:
for(i = n; i < 2*n; ++i) {
first[i] = first[i-1] + nums[i];
maxq.push( nums[i] );
int maxv = maxq.top();
maxq.pop();
first[i] -= maxv;
}
(
6
)
(6)
(6) 对于右边的部分也是同样计算,但是右边的部分不需要存入数组,每次计算完毕直接和 first[i]
进行差值,然后去最小值即可。
2、时间复杂度
最坏时间复杂度 O ( n l o g n ) O(nlogn) O(nlogn) 。
3、代码详解
class Solution {
public:
long long minimumDifference(vector<int>& nums) {
int i;
priority_queue<int, vector<int>, less<int>> maxq;
priority_queue<int, vector<int>, greater<int>> minq;
int n = nums.size() / 3;
long long first[200010];
long long ret = 1000000000;
ret = ret * ret;
first[n-1] = 0;
for(i = 0; i < n; ++i) {
first[n-1] += nums[i];
maxq.push( nums[i] );
}
for(i = n; i < 2*n; ++i) {
first[i] = first[i-1] + nums[i];
maxq.push( nums[i] );
int maxv = maxq.top();
maxq.pop();
first[i] -= maxv;
}
long long sec = 0;
for(i = 3*n-1; i >= 2*n; --i) {
sec += nums[i];
minq.push( nums[i] );
}
ret = first[2*n-1] - sec;
for(i = 2*n-1; i >= n; --i) {
sec += nums[i];
minq.push( nums[i] );
int minv = minq.top();
minq.pop();
sec -= minv;
ret = min(ret, first[i-1] - sec);
}
return ret;
}
};
三、本题小知识
窗口是可以滑,堆是可以和滑动窗口配合使用的。
四、加群须知
相信看我文章的大多数都是「 大学生 」,能上大学的都是「 精英 」,那么我们自然要「 精益求精 」,如果你还是「 大一 」,那么太好了,你拥有大把时间,当然你可以选择「 刷剧 」,然而,「 学好算法 」,三年后的你自然「 不能同日而语 」。
那么这里,我整理了「 几十个基础算法 」 的分类,点击开启:
如果链接被屏蔽,或者有权限问题,可以私聊作者解决。
大致题集一览:
为了让这件事情变得有趣,以及「 照顾初学者 」,目前题目只开放最简单的算法 「 枚举系列 」 (包括:线性枚举、双指针、前缀和、二分枚举、三分枚举),当有 一半成员刷完 「 枚举系列 」 的所有题以后,会开放下个章节,等这套题全部刷完,你还在群里,那么你就会成为「 夜深人静写算法 」专家团 的一员。
不要小看这个专家团,三年之后,你将会是别人 望尘莫及 的存在。如果要加入,可以联系我,考虑到大家都是学生, 没有「 主要经济来源 」,在你成为神的路上,「 不会索取任何 」。
🔥联系作者,或者扫作者主页二维码加群,加入刷题行列吧🔥
🔥让天下没有难学的算法🔥
C语言免费动漫教程,和我一起打卡! 🌞《光天化日学C语言》🌞
让你养成九天持续刷题的习惯 🔥《九日集训》🔥
入门级C语言真题汇总 🧡《C语言入门100例》🧡
组团学习,抱团生长 🌌《算法零基础100讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜