LeetCode 1537.最大得分

发布于:2025-09-05 ⋅ 阅读:(12) ⋅ 点赞:(0)

你有两个 有序 且数组内元素互不相同的数组 nums1 和 nums2 。

一条 合法路径 定义如下:

选择数组 nums1 或者 nums2 开始遍历(从下标 0 处开始)。
从左到右遍历当前数组。
如果你遇到了 nums1 和 nums2 中都存在的值,那么你可以切换路径到另一个数组对应数字处继续遍历(但在合法路径中重复数字只会被统计一次)。
得分 定义为合法路径中不同数字的和。

请你返回 所有可能 合法路径 中的最大得分。由于答案可能很大,请你将它对 10^9 + 7 取余后返回。

示例 1:
在这里插入图片描述
输入:nums1 = [2,4,5,8,10], nums2 = [4,6,8,9]
输出:30
解释:合法路径包括:
[2,4,5,8,10], [2,4,5,8,9], [2,4,6,8,9], [2,4,6,8,10],(从 nums1 开始遍历)
[4,6,8,9], [4,5,8,10], [4,5,8,9], [4,6,8,10] (从 nums2 开始遍历)
最大得分为上图中的绿色路径 [2,4,6,8,10] 。
示例 2:

输入:nums1 = [1,3,5,7,9], nums2 = [3,5,100]
输出:109
解释:最大得分由路径 [1,3,5,100] 得到。
示例 3:

输入:nums1 = [1,2,3,4,5], nums2 = [6,7,8,9,10]
输出:40
解释:nums1 和 nums2 之间无相同数字。
最大得分由路径[6,7,8,9,10]得到。

提示:

1 <= nums1.length, nums2.length <= 10 5 ^5 5
1 <= nums1[i], nums2[i] <= 10 7 ^7 7
nums1 和 nums2 都是严格递增的数组。

双序列双指针,找两个数组中相同值的节点,并计算到相同值的两条路径的得分,取较大者贪心即可:

class Solution {
public:
    int maxSum(vector<int>& nums1, vector<int>& nums2) {
        int idx1 = 0;
        int idx2 = 0;

        long long path1 = 0;
        long long path2 = 0;

        long long ans = 0;

        while (idx1 < nums1.size() && idx2 < nums2.size()) {
            // 移动两者中较小的那个,以找到相同值
            if (nums1[idx1] < nums2[idx2]) {
                path1 += nums1[idx1];
                ++idx1;
            } else if (nums1[idx1] > nums2[idx2]) {
                path2 += nums2[idx2];
                ++idx2;
            } else {
                // 找到相同值后,取较大得分的路径
                ans += max(path1, path2) + nums1[idx1];

                // 初始化下一段路径的得分
                path1 = 0;
                path2 = 0;

                ++idx1;
                ++idx2;
            }
        }

        while (idx1 < nums1.size()) {
            path1 += nums1[idx1];
            ++idx1;
        }

        while (idx2 < nums2.size()) {
            path2 += nums2[idx2];
            ++idx2;
        }

        ans += max(path1, path2);

        return ans % (long long)(1e9 + 7);
    }
};

如果nums1的长度为n,nums2的长度为m,则此算法时间复杂度为O(n+m),空间复杂度为O(1)。


网站公告

今日签到

点亮在社区的每一天
去签到