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 <= nums1[i], nums2[i] <= 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)。

相关推荐
MM_MS9 小时前
Halcon变量控制类型、数据类型转换、字符串格式化、元组操作
开发语言·人工智能·深度学习·算法·目标检测·计算机视觉·视觉检测
独自破碎E10 小时前
【二分法】寻找峰值
算法
mit6.82410 小时前
位运算|拆分贪心
算法
ghie909010 小时前
基于MATLAB的TLBO算法优化实现与改进
开发语言·算法·matlab
恋爱绝缘体110 小时前
2020重学C++重构你的C++知识体系
java·开发语言·c++·算法·junit
wuk99810 小时前
VSC优化算法MATLAB实现
开发语言·算法·matlab
Z1Jxxx11 小时前
加密算法加密算法
开发语言·c++·算法
乌萨奇也要立志学C++11 小时前
【洛谷】递归初阶 三道经典递归算法题(汉诺塔 / 占卜 DIY/FBI 树)详解
数据结构·c++·算法
vyuvyucd11 小时前
C++引用:高效编程的别名利器
算法
鱼跃鹰飞12 小时前
Leetcode1891:割绳子
数据结构·算法