用于解决一类基于子段的统计问题
子段就是:数组中连续的一段
可以用一个
闭区间
来表示数组中的连续一段
这个方法核心就是优化:两种循环的枚举
就是根据题意,把[j,i]范围中,j的这层循环去掉(j从0~i)
就是滑动窗口,一个端点跟随另一个端点来移动
sort(first_pointer,first_pointer+n,cmp)
int a[8] = {6, 7, 2, 9, 1, 3, 5, 2};
sort(a, a + 8,less());//less<数据类型>()
//默认从小到大,less()可省略
int a[8] = {6, 7, 2, 9, 1, 3, 5, 2};
sort(a, a + 8,greater());
//这是从大到小排列
注意:less<>()和greater<>()数据类型要写对
float f[5] = {11.2, 76.7, 8.32, 15.12, 2.676}; sort(f, f+sizeof(f)/sizeof(float),less
()); double d[5] = {1.2, 7.7, 8.32, 5.12, 2.66}; sort(d, d + 5,greater ());
vector一样,而且传参可以传入自带的迭代器
vector
f= {11.2, 76.7, 8.32, 15.12, 2.676}; sort(f.begin(),f.end(),less ()); vector d= {1.2, 7.7, 8.32, 5.12, 2.66}; sort(d.begin(), d.end(),greater ());
sort()第三个参数可以自己自定义,注意函数要用bool返回值,而且传参时不要加()
#include
#include
#include
#include
using namespace std;
struct student{
string name;
int age;
student(string s,int a):name(s),age(a){}
};
//按姓名升序
bool cmp_name(const student& a,const student& b){
return a.name < b.name;
}
//按年龄降序
bool cmp_age(const student& a,const student& b){
return a.age > b.age;
}
int main(){
student stu[5] = {
{"daming", 54},
{"xiaozhang", 44},
{"amy", 33},
{"bob", 12},
{"alicc", 77}};
sort(stu, stu+5, cmp_name);
for(auto [name,age]:stu){
cout << name << "," << age << endl;
}
vector vec(stu, stu + 5);//迭代器初始化
sort(vec.begin(), vec.end(), cmp_age);
for(auto [name,age]:vec){
cout << name << "," << age << endl;
}
}
数组指针初始化
vector vec(
stu + 1
,stu + 4
);
参数的函数无括号
sort(vec.begin(), vec.end(), cmp_age);
167. 两数之和 II - 输入有序数组
利用排序特点,一端指向最小,一段指向最大。那么相加多了,大的减小,少了,小的增加。
vector twoSum(vector& numbers, int target) {
//左右指针向中间移动
int i=0;
int j=numbers.size()-1;
while(itarget){
j--;
}
else if(numbers[i]+numbers[j]
法二:固定一端,考虑另一端的变化情况
vector twoSum(vector& numbers, int target) {
//左右指针向中间移动
int j=numbers.size()-1;
for(int i=0;itarget) j--;
if(i
这里{
i+1,j+1
}是题目要求从1开始只有:
i
这个下标才有意义,所以要加入判断 或者在代码写:if(i>=j) break;
1. 两数之和
采用自定义类型来存储原数组下标,然后排序
注意自定义类型的访问名称是类型的变量名称
vec[i].val
vec[i].index
- 这里cmp函数需要加上static关键字
class Solution {
public:
vector twoSum(vector& nums, int target) {
int n = nums.size();
vector- vec;
for (int i = 0; i < n; i++) {
//用vec存储排序前的下标
vec.push_back(item(nums[i], i));
}
sort(vec.begin(),vec.end(),cmp);
int j = n - 1;
for (int i = 0; i < n; i++) {
while (i < j && vec[i].val + vec[j].val > target)
j--;
if (i < j && vec[i].val + vec[j].val == target) {
return {vec[i].index, vec[j].index};
}
}
return {};
}
private:
struct item {
int val;
int index;
item(int v, int i) : val(v), index(i) {}
};
static bool cmp(const item& a, const item& b) { return a.val < b.val; }
};
或者直接用现成的pair,注意first和second是pair的成员
vector twoSum(vector& nums, int target) {
//用pair来存储
vector> vec;
for(int i=0;itarget){
j--;
}
if(i
15. 三数之和
双指针法的优势就是可以找到所有可能解,因为排序后的数组只要一大一小向中间移动,就会找到多个解
vector> colTwoSum(vector &nums, int target){
vector> res;
int k = nums.size() - 1;
for (int j=0; j < nums.size(); j++){
while (j < k && nums[j] + nums[k] > target)
k--;
if (j < k && nums[j] + nums[k] == target){
//只存不return,会找到所有结果
res.push_back({nums[j], nums[k]});
}
}
return res;
}
int main(){
vector nums = {1, 3, 4, 5, 6, 7, 8};
auto res = colTwoSum(nums, 9);
for(auto v:res){
cout << v[0] << " " << v[1] << endl;
}
}
/*
1 8
3 6
4 5*/
三数之和就是在两数之和基础上改进,关键:nums[j]+nums[k]==-nums[i],所以就是遍历i然后利用双指针的两数之和找到所有满足的j,k,然后把当前i并入结果即可
class Solution {
public:
vector> threeSum(vector& nums) {
//题目要求返回值,所以可以直接排序
sort(nums.begin(),nums.end());
//遍历nums[i],向后找:nums[j]和nums[k]
//即:nums[j]+nums[k]==-nums[i](保证i,j,k不同)
vector> ans;
for(int i=0;i=1 && nums[i]==nums[i-1]) continue;
auto res=colTwoSum(nums,i+1,-nums[i]);
//所有的res结果并入nums[i]
for(auto v:res){
ans.push_back({nums[i],v[0],v[1]});
}
}
return ans;
}
private:
vector> colTwoSum(vector& nums,int j,int target){
vector> res;
int k=nums.size()-1;
int tmp=j;
for(;jtmp && nums[j]==nums[j-1]) continue;
while(jtarget) k--;
if(j
也可以不写函数:
vector> threeSum(vector& nums) {
//题目要求返回值,所以可以直接排序
sort(nums.begin(),nums.end());
//遍历nums[i],向后找:nums[j]和nums[k]
//即:nums[j]+nums[k]==-nums[i](保证i,j,k不同)
vector> ans;
for(int i=0;i=1 && nums[i]==nums[i-1]) continue;
int k=nums.size()-1;//类似函数传参,初始值定义在外面
int target=-nums[i];
for(int j=i+1;ji+1 && nums[j]==nums[j-1]) continue;
while(jtarget) k--;
if(j
11. 盛最多水的容器
解题关键就是意识到:面积是由短板和距离决定,如果距离最长,那面积完全取决于短板
所以每次计算面积后,移动最短的那块板
这里i,j两端变化无规律,取决于height[i]和height[j],用while最合适
int maxArea(vector& height) {
int res=0;
//双指针
int i=0,j=height.size()-1;
while(i
res=height[i]
注意:因为自增自减,不可以用:height[i++]*(j-i);
单调递增堆栈
是指元素从下到上按升序排列的堆栈
单调递减堆栈
是元素从底部到顶部按降序排列的堆栈
//栈顶元素最大,而且出栈大小依次递减
考虑一个数组 Arr[] = {1, 4, 5, 3, 12, 10}
对于 i = 0: stk = {1}
对于 i = 1: stk = {1, 4}
对于 i = 2: stk = { 1, 4, 5}
对于 i = 3: stk = {1, 3} [弹出 4 和 5 作为 4 > 3 和 5 > 3]
对于 i = 4: stk = {1, 3, 12}
对于 i = 5 : stk = {1, 3, 10} [将 12 弹出为 12 > 10]
class monoStack{
public:
/*单调递增栈*/
void increase_stack(vector vec){
//初始化一个栈,存下标
stack sta;
//访问每一个元素
for (int i = 0; i < vec.size();i++){
//新元素一定会入栈,但是必须把大于它的值出栈
while(!sta.empty()&& vec[i]= 0;j--){
res[j] = vec[sta.top()];
sta.pop();
}
}
private:
vector res;
};
int main(){
vector a{1, 4, 5, 3, 12, 10};
monoStack ms;
ms.increase_stack(a);
}
递减单调栈(栈底元素最大,栈顶元素最小)
void decrease_stack(vector vec){
//递减,栈底元素最大
stack sta;
for (int i = 0; i < vec.size();i++){
//入栈前,必须去掉栈中比自己小的
while(!sta.empty()&&sta.top()= 0;i--){
res[i] = sta.top();
sta.pop();
}
}
84. 柱状图中最大的矩形
题目特点就是:可延伸,是直方图内部的构成的最大面积
如果用暴力解法:
+ 左右指针初始化为当前柱子i,高度:heights[i]
+ 只要两端左边或右边比heght[i]小就是可延伸
+ 也就是:从中心向两端扩散
int largestRectangleArea(vector& heights) {
int n=heights.size();
int res=0;
for(int i=0;i=0 && heights[left]>=h) left--;
/*
for(;left>=0;left--)
if(heghts[left]=h) right++;
int w=right-left-1;//延伸宽度:(left,right)
res=max(res,w*h);
}
return res;
}
应用单调栈
特点就是:每次遍历柱子的时候,如果当前柱子高度比他的前一个柱子矮的话(比如:1比0矮)
此时:前一个柱子高度一定不能向后扩展,所以会计算出以它的高度可能向前延申的面积
这里的123是严格递增的,所以此时他们各自高度形成最大面积不能判断,因为可能会向后延申
也就是说,每次我们都把单调不减的柱子放入单调栈,万一当前柱子比栈顶元素高度小,那么就出栈,可以计算出它的面积了
问题关键就是:向前延申的面积的宽度怎么确定?
就是:当前出栈元素高度height[sta.top()],出栈后,当前访问访问柱子i和单调栈栈顶元素之间的宽度
也就是:i-栈top()位置-1
因为:我们维持一个单调递增的单调栈,如果柱子i比栈顶小,那么前一个一定可以计算出面积。
而它延申的最长距离就是出栈后的单调栈栈顶位置(因为严格单调递增一定延申不到这个位置)
注意到,访问最后一个元素后没有结束,所以需要再加入一个小值,以便把整个栈内元素出栈
int largestRectangleArea(vector& heights) {
int n=heights.size();
int res=0;
stack sta;
for(int i=0;i<=n;i++){
//为了让最后一个元素能出栈,不妨让n时直方图高度-1
//-1会保证单调栈内所以元素出栈
int curHeight=(i==n)?-1:heights[i];
/*单调栈模板:先把破坏单调性元素出栈,然后必入栈*/
while(!sta.empty() && curHeight<=heights[sta.top()]){
//i-1位置的面积可求
int h=heights[sta.top()];
sta.pop();//计算面积后就不存了
//因为sta.top()在栈空时不存在
int w;//宽度
if(!sta.empty()) w=i-sta.top()-1;
else w=i;//0 1 .. 这根柱子(i-1) i---(i-1)-0+1
res=max(res,w*h);
}
sta.push(i);//必入栈
}
return res;
}
curHeight<=heights[sta.top()]
这里考虑了相等柱子不入栈,跟测试用例有关,写<也行
if(!sta.empty()) w=i-sta.top()-1;
else w=i;//0 1 .. 这根柱子(i-1) i---(i-1)-0+1
由于递减形状可能会让栈没元素,注意:sta.top()的存在前提是栈不空
单调栈需要考虑两种特殊的情况:
在输入数组的两端加上两个高度为 0的柱形,可以回避上面这两种分类讨论。
这两个站在两边的柱形有一个很形象的名词,叫做哨兵(Sentinel)
有了这两个柱形:
左边的柱形(第 1 个柱形)由于它一定比输入数组里任何一个元素小,它肯定不会出栈,因此栈一定不会为空;
右边的柱形(第 2 个柱形)也正是因为它一定比输入数组里任何一个元素小,它会让所有输入数组里的元素出栈(第 1 个哨兵元素除外)。
int largestRectangleArea(vector& heights) {
//在原来直方图前后两端各加一个0
//右端的0会让单调栈全部出栈
//左端的0会让单调栈不为空
heights.insert(heights.begin(),0);
heights.insert(heights.end(),0);
int res=0;
stack sta;
for(int i=0;i