LeetCode第271场周赛题解

2021.12.12LeetCode第271场周赛

2103. 环和杆

思路

使用位运算记录状态
每个int的三位表示三种颜色 包含三种颜色的值为7

代码

class Solution {
public:
    int countPoints(string rings) {
        vector<int> state(10);
        unordered_map<char, int> mp = {{'R', 1}, {'G', 2}, {'B', 4}};
        for (int i = 0; i < rings.size(); i += 2) {
            state[rings[i + 1] - '0'] |= mp[rings[i]];
        }
        int ans = 0;
        for (int i = 0; i < 10; i ++ )
            if (state[i] == 7)
                ans ++ ;
        return ans;
    }
};

2104. 子数组范围和

思路

固定左端点 枚举右端点
动态维护当前区间内的最大值和最小值
该做法的时间复杂度为 $O(n^2)$
$O(n)$ 的做法请点击这里

代码

class Solution {
public:
    long long subArrayRanges(vector<int>& nums) {
        int n = nums.size();
        long long ans = 0;
        for (int l = 0; l < n; l ++ ) {
            int a = nums[l], b = nums[l];
            for (int r = l; r < n; r ++ ) {
                a = min(a, nums[r]);
                b = max(b, nums[r]);
                ans += (long long)(b - a);
            }
        }
        return ans;
    }
};

2105. 给植物浇水 II

思路

双指针模拟
两人同时从最左边和最右边开始浇水 若无法浇水则加满
最后如果相遇处理特殊情况

代码

class Solution {
public:
    int minimumRefill(vector<int>& p, int ca, int cb) {
        int n = p.size();
        int i = 0, j = n - 1;
        int ans = 0;
        int a = ca, b = cb;
        while (i < j) {
            if (a < p[i])
                ans ++ , a = ca;
            if (b < p[j])
                ans ++ , b = cb;
            a -= p[i];
            b -= p[j];
            i ++ , j -- ;
        }
        if (i == j) {
            if (a == b || a > b) {
                if (a < p[i])
                    ans ++ ;
            }
            else {
                if (b < p[i])
                    ans ++ ;
            }
        }
        return ans;
    }
};

2106. 摘水果

思路

前缀和+贪心
先预处理出所有位置的前缀和 注意范围区间是 $[0,2\times 10^5]$
为方便计算下标从1开始
为使摘到的水果最多 应尽可能多走距离
故只存在两种情况 先向左走 再向右走 和 先向右走 再向左走
枚举所有情况的最大和即可

代码

class Solution {
public:
    int maxTotalFruits(vector<vector<int>>& f, int startPos, int k) {
        int n = f.size();
        int N = 2 * 1e5 + 10;
        vector<int> s(N + 1);
        int j = 0;
        for (int i = 1; i <= N; i ++ ) {
            s[i] = s[i - 1];
            if (j < n && i == f[j][0] + 1)
                s[i] = s[i - 1] + f[j][1], j ++ ;
        }
        
        int ans = 0;
        startPos ++ ;
        for (int i = 0; i <= k; i ++ ) {
            int l = max(1, startPos - i); //先向左走i
            int r = min(N, startPos - i + (k - i)); //再向右走k - i
            if (r < startPos) r = startPos;
            if (l > startPos) l = startPos;
            ans = max(ans, s[r] - s[l - 1]);
            
            r = min(N, startPos + i); //先向右走i
            l = max(1, startPos + i - (k - i)); //再向左走k - i
            if (r < startPos) r = startPos;
            if (l > startPos) l = startPos;
            ans = max(ans, s[r] - s[l - 1]);
        }
        return ans;
    }
};