Leetcode 第 137 场双周赛题解

Leetcode 第 137 场双周赛题解

题目1：3254. 长度为 K 的子数组的能量值 I

思路

遍历数组 nums 中的每个长度为 k 的子数组，判断子数组中的所有相邻元素是否满足后一个元素与前一个元素之差等于 1，如果满足则子数组的能量值等于子数组的最后一个元素，否则子数组的能量值等于 −1。

代码

/*
 * @lc app=leetcode.cn id=3254 lang=cpp
 *
 * [3254] 长度为 K 的子数组的能量值 I
 */

// @lc code=start
class Solution
{
public:
    vector<int> resultsArray(vector<int> &nums, int k)
    {
        int n = nums.size();
        vector<int> ans;
        for (int i = 0; i <= n - k; i++)
        {
            bool flag = true;
            for (int j = i + 1; j < i + k && flag; j++)
                if (nums[j] != nums[j - 1] + 1)
                    flag = false;
            if (flag)
                ans.push_back(nums[i + k - 1]);
            else
                ans.push_back(-1);
        }
        return ans;
    }
};
// @lc code=end

复杂度分析

时间复杂度：O(n * k)，其中 n 是数组 nums 的长度，k 是给定的正整数。子数组的数量是 n−k+1，对于每个子数组计算能量值的时间是 O(k)，因此时间复杂度是 O(n * k)。

空间复杂度：O(1)。注意返回值不计入空间复杂度。

题目2：3255. 长度为 K 的子数组的能量值 II

思路

遍历数组的同时，用一个计数器 cnt 统计连续递增的元素个数：

• 初始化 cnt=0。
• 如果 i=0 或者 nums[i]=nums[i−1]+1，则把 cnt 增加 1。
• 否则，把 cnt 置为 1。

如果 cnt≥k，则更新 ans[i−k+1]=nums[i]。

代码

/*
 * @lc app=leetcode.cn id=3255 lang=cpp
 *
 * [3255] 长度为 K 的子数组的能量值 II
 */

// @lc code=start
class Solution
{
public:
    vector<int> resultsArray(vector<int> &nums, int k)
    {
        int n = nums.size();
        int cnt = 0; // 连续递增的元素个数
        vector<int> ans(n - k + 1, -1);
        for (int i = 0; i < n; i++)
        {
            if (i == 0 || nums[i] == nums[i - 1] + 1)
                cnt++;
            else
                cnt = 1;
            if (cnt >= k)
                ans[i - k + 1] = nums[i];
        }
        return ans;
    }
};
// @lc code=end

复杂度分析

时间复杂度：O(n)，其中 n 是数组 nums 的长度。

空间复杂度：O(1)。返回值不计入。

题目3：3256. 放三个车的价值之和最大 I

思路

从大到下排序矩阵所有值, 记为数组v。

转化此题：从r*c个数中选取3个数分别给到车1，车2，和车3，使得符合条件的三数之和最大。

结论：可选前2r+2c+1个最大值作为候选值，在此范围枚举求三数之和最大。

代码

/*
 * @lc app=leetcode.cn id=3256 lang=cpp
 *
 * [3256] 放三个车的价值之和最大 I
 */

// @lc code=start
class Solution
{
public:
    long long maximumValueSum(vector<vector<int>> &board)
    {
        int m = board.size(), n = m ? board[0].size() : 0;
        vector<tuple<int, int, int>> v;
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                v.push_back({board[i][j], i, j});
        sort(v.begin(), v.end(), greater<tuple<int, int, int>>());

        long long ans = LONG_LONG_MIN;
        int range = min(2 * (m + n) + 1, (int)v.size());
        for (int i = 0; i < range; i++)
        {
            auto [v1, x1, y1] = v[i];
            for (int j = i + 1; j < range; j++)
            {
                auto [v2, x2, y2] = v[j];
                if (x2 == x1 || y2 == y1)
                    continue;
                for (int k = j + 1; k < range; k++)
                {
                    auto [v3, x3, y3] = v[k];
                    if (x3 == x1 || y3 == y1 || x3 == x2 || y3 == y2)
                        continue;
                    ans = max(ans, (long long)v1 + v2 + v3);
                }
            }
        }
        return ans;
    }
};
// @lc code=end

复杂度分析

时间复杂度：O((m+n)³)，其中 m 和 n 分别是数组 board 的行数和列数。

空间复杂度：O(m * n)，其中 m 和 n 分别是数组 board 的行数和列数。

题目4：3257. 放三个车的价值之和最大 II

思路

前后缀分解。

题解：https://leetcode.cn/problems/maximum-value-sum-by-placing-three-rooks-ii/solutions/2884186/qian-hou-zhui-fen-jie-pythonjavacgo-by-e-gc48/

代码

/*
 * @lc app=leetcode.cn id=3257 lang=cpp
 *
 * [3257] 放三个车的价值之和最大 II
 */

// @lc code=start
class Solution
{
public:
    long long maximumValueSum(vector<vector<int>> &board)
    {
        array<pair<int, int>, 3> p; // 最大、次大、第三大
        ranges::fill(p, make_pair(INT_MIN, -1));

        auto update = [&](vector<int> &row)
        {
            for (int j = 0; j < row.size(); j++)
            {
                int x = row[j];
                if (x > p[0].first)
                {
                    if (p[0].second != j)
                    { // 如果相等，仅更新最大
                        if (p[1].second != j)
                        { // 如果相等，仅更新最大和次大
                            p[2] = p[1];
                        }
                        p[1] = p[0];
                    }
                    p[0] = {x, j};
                }
                else if (x > p[1].first && j != p[0].second)
                {
                    if (p[1].second != j)
                    { // 如果相等，仅更新次大
                        p[2] = p[1];
                    }
                    p[1] = {x, j};
                }
                else if (x > p[2].first && j != p[0].second && j != p[1].second)
                {
                    p[2] = {x, j};
                }
            }
        };

        int m = board.size(), n = board[0].size();
        vector<array<pair<int, int>, 3>> suf(m);
        for (int i = m - 1; i > 1; i--)
        {
            update(board[i]);
            suf[i] = p;
        }

        long long ans = LLONG_MIN;
        ranges::fill(p, make_pair(INT_MIN, -1)); // 重置，计算 pre
        for (int i = 1; i < m - 1; i++)
        {
            update(board[i - 1]);
            for (int j2 = 0; j2 < n; j2++)
            { // 第二个车
                for (auto &[x, j1] : p)
                { // 第一个车
                    for (auto &[z, j3] : suf[i + 1])
                    { // 第三个车
                        if (j1 != j2 && j1 != j3 && j2 != j3)
                        { // 没有同列的车
                            ans = max(ans, (long long)x + board[i][j2] + z);
                            break;
                        }
                    }
                }
            }
        }
        return ans;
    }
};
// @lc code=end

复杂度分析

时间复杂度：O(m * n)，其中 m 和 n 分别是数组 board 的行数和列数。

空间复杂度：O(m * n)，其中 m 是数组 board 的行数。