LeetCode刷題記錄——day4

想成为编程高手的阿曼發表於2024-03-22

https://leetcode.cn/problems/trapping-rain-water/description/?envType=study-plan-v2&envId=top-interview-150
對於一個可以構成“碗”的序列,最後裝滿水的話應該和最短的一邊齊平,那麼可以左右各遍歷一次,記錄每個元素位置對應的最短邊高度,再對比就可以得出左右哪邊最短

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        if (n == 0) {
            return 0;
        }
        vector<int> leftMax(n);
        leftMax[0] = height[0];
        for (int i = 1; i < n; ++i) {
            leftMax[i] = max(leftMax[i - 1], height[i]);
        }

        vector<int> rightMax(n);
        rightMax[n - 1] = height[n - 1];
        for (int i = n - 2; i >= 0; --i) {
            rightMax[i] = max(rightMax[i + 1], height[i]);
        }

        int ans = 0;
        for (int i = 0; i < n; ++i) {
            ans += min(leftMax[i], rightMax[i]) - height[i];
        }
        return ans;
    }
};

總結,對於一些左右對稱的問題,也就是將其對稱翻轉過來結果一樣的問題,或者說有時我們需要同時知道左右元素的資訊才能得到答案的,我們不可能只從一邊遍歷一次就得到答案,應該左右各遍歷一次才能同時得到左邊的資訊和右邊的資訊,這樣才能夠可能得到答案

本文由部落格一文多發平臺 OpenWrite 釋出!

相關文章