LeetCode || 双指针 / 单调栈

11. Container With Most Water

题意:取两根求最大体积

思路:使用两个指针分别指向头和尾,然后考虑左右两根:

对于小的那根,如果选择了它,那么能够产生的最大体积一定是当前的情况:因为当前情况体积=这根长度 * 当前距离,以后的话长度一定是左右边界的min,即<=这根长度,距离一定越来越小

因此小的那根以后就不会被选了,因此把小的那根的指针挪动一下

直到指针相遇

class Solution {
public:
    int maxArea(vector<int>& height) {
        int ans = 0;
        int l = 0, r = height.size() - 1;
        while (l < r) {
            ans = max(ans, (r - l) * min(height[l], height[r]));
            if (height[l] <= height[r]) l++;
            else r--;
        }
        return ans;
    }
};
View Code

42. Trapping Rain Water

题意:给一些柱形块,求下雨时可积水的体积,如图

思路:用一个单调递减栈,只有当遇到更高的柱形的时候才会产生积水,此时的栈顶是积水的底,次栈顶是左边界,当前是右边界。

每次先栈中把比当前柱小的都计算好面积出栈,然后将当前柱入栈

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int> s;
        int n = height.size();
        int ans = 0;
        for (int i = 0; i < n; i++) {
            if (s.empty() || height[i] < height[s.top()]) {
                s.push(i);
            }
            else {
                while (!s.empty() && height[s.top()] < height[i]) {
                    int t = s.top();
                    s.pop();
                    if (!s.empty()) {
                        ans += (i - s.top() - 1) * (min(height[s.top()], height[i]) - height[t]);
                    }
                }
                s.push(i);
            }
        }
        return ans;
    }
};
View Code

15. 3Sum

题意:给一组数,取三个数使它们和为0,输出这三个数,要求去重

思路:On3枚举超时,排序后,先取定其中一个数nums[i],然后两个指针分别指向 i 之后的第一个和最后一个数字,由于数字是递增的,当 i, j, k 的和<0时 j++,和>0时 k--

然后当有重复时指针继续移动即可

vector中的一些用法

//声明vector
vector<int> v;
vector<vector<int>> v;
//输入
v.push_back(a);
v.push_back({a, b, c});
//排序
sort(v.begin(), v.end());
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        int n = nums.size();
        int idx = 0;
        vector<vector<int>> ans;
        for (int i = 0; i < n; i++) {
            if (i && nums[i] == nums[i - 1]) continue;
            int j = i + 1, k = n - 1;
            while (j < k) {
                if (nums[i] + nums[j] + nums[k] < 0) {
                    j++;
                } else if (nums[i] + nums[j] + nums[k] > 0) {
                    k--;
                } else {
                    ans.push_back({nums[i], nums[j], nums[k]});
                    while (j < n - 1 && nums[j] == nums[j + 1]) j++;
                    while (k > 0 && nums[k] == nums[k - 1]) k--;
                    j++; k--;
                }
            }
        }
        return ans;
    }
};
View Code

16. 3Sum Closest

题意:一组数,取三个数使它们的和与target的差最小

思路:还是双指针的思路,先取定一个数

排好序之后用双指针可以让On2的遍历求和降成On

为确定哪个差最小 dif 里面存差,ans存结果

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        int n = nums.size();
        int dif = 1e9, ans = 0;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < n; i++) {
            int l = i + 1, r = n - 1;
            while (l < r) {
                int sum = nums[i] + nums[l] + nums[r];
                if (abs(target - sum) < dif) {
                    dif = abs(target - sum);
                    ans = sum;
                }
                if (sum < target) {
                    l++;
                } else if (sum > target) {
                    r--;
                } else return target;
            }
        }
        return ans;
    }
};
View Code
原文地址:https://www.cnblogs.com/pinkglightning/p/10317384.html