【排序-】1508. 子数组和排序后的区间和

最新推荐文章于 2022-04-27 15:08:26 发布

原创最新推荐文章于 2022-04-27 15:08:26 发布 · 108 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python

leetcode 专栏收录该内容

714 篇文章

订阅专栏

博客给出一个数组相关的算法问题，需计算所有非空连续子数组的和并排序，返回新数组中指定下标区间的数字和。给出多个示例，同时给出了问题的输入提示，最后提到使用Python解决该问题。

【题目】
给你一个数组 nums ，它包含 n 个正整数。你需要计算所有非空连续子数组的和，并将它们按升序排序，得到一个新的包含 n * (n + 1) / 2 个数字的数组。
请你返回在新数组中下标为 left 到 right （下标从 1 开始）的所有数字和（包括左右端点）。由于答案可能很大，请你将它对 10^9 + 7 取模后返回。
【示例 1】
输入：nums = [1,2,3,4], n = 4, left = 1, right = 5
输出：13
解释：所有的子数组和为 1, 3, 6, 10, 2, 5, 9, 3, 7, 4 。将它们升序排序后，我们得到新的数组 [1, 2, 3, 3, 4, 5, 6, 7, 9, 10] 。下标从 le = 1 到 ri = 5 的和为 1 + 2 + 3 + 3 + 4 = 13 。
【示例 2】
输入：nums = [1,2,3,4], n = 4, left = 3, right = 4
输出：6
解释：给定数组与示例 1 一样，所以新数组为 [1, 2, 3, 3, 4, 5, 6, 7, 9, 10] 。下标从 le = 3 到 ri = 4 的和为 3 + 3 = 6 。
【示例 3】
输入：nums = [1,2,3,4], n = 4, left = 1, right = 10
输出：50
【提示】
1 <= nums.length <= 10^3
nums.length == n
1 <= nums[i] <= 100
1 <= left <= right <= n * (n + 1) / 2
【代码】
【Python】

class Solution:
    def deal(self,s):
        for i in range(1,len(s)):
            s[i]+=s[i-1]
        return s
    def rangeSum(self, nums: List[int], n: int, left: int, right: int) -> int:
        rs=[]
        for i in range(n):
            rs.extend(self.deal(nums[i:]))
        rs.sort()
        num=0
        temp=pow(10,9)+7
        for i in range(left-1,right):
            num+=rs[i]
            num%=temp            
        return num