LeetCode 1537.最大得分

你有两个 有序 且数组内元素互不相同的数组 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 <= 105^55

1 <= nums1i, nums2i <= 107^77

nums1 和 nums2 都是严格递增的数组。

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

cpp 复制代码
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)。

相关推荐
小欣加油4 分钟前
leetcode1926 迷宫中离入口最近的出口
数据结构·c++·算法·leetcode·职场和发展
happymaker06262 小时前
LeetCodeHot100——42.接雨水
算法
阿正的梦工坊3 小时前
【Rust】07-错误处理:Option、Result 与 ? 运算符
开发语言·算法·rust
八解毒剂4 小时前
数据结构-平衡二叉树——对二叉搜索树的优化
数据结构·c++·算法
运行时记录5 小时前
别再手动写提示词了 — SkillOpt 让技能文档自己进化
算法
啦啦啦啦啦zzzz5 小时前
算法总结(二分查找、双指针)
c++·算法
qq_8573058196 小时前
python语法
开发语言·python·算法
DXM05216 小时前
第9期|从机器学习到深度学习:AI遥感解译的进化逻辑
人工智能·算法·计算机视觉
小蒋学算法6 小时前
算法-阶乘函数后K个零
算法
weixin_307779136 小时前
智能模拟数据生成平台:生成式AI合成数据技术重塑开发测试效能
人工智能·测试工具·算法·测试用例