LeetCode-Python-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] 得到。
示例 4：

输入：nums1 = [1,4,5,8,9,11,19], nums2 = [2,3,4,11,12]
输出：61
 

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/get-the-maximum-score
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

此题难点在于不要被困难的标签迷惑了。

思路非常简单，以交点为界限将数组分为不同的段。

如果有N个交点，那么每个数组都可以分成N + 1段。

对于每段，可以选择走nums1， 也可以选择走nums2，

如果每段都走较大的那条路，那么总结果必定最大。

时间复杂度：O（M + N）

空间复杂度：O（M + N），可以优化到O（1）

class Solution(object):
    def maxSum(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: int
        """
        dups = set(nums1) & set(nums2)
        # 求分段和数组
        s1 = self.getFragmentedSum(nums1, dups)
        s2 = self.getFragmentedSum(nums2, dups)

        res = 0
        for sum1, sum2 in zip(s1, s2):
            # 每段取较大和
            res += max(sum1, sum2)
        return res % (10 ** 9 + 7)

    def getFragmentedSum(self, nums, dups):
        l = []
        s = 0
        for num in nums:
            s += num 
            if num in dups:
                l.append(s)
                s = 0
        l.append(s)
        return l