题解 | #计算数组的小和#

计算数组的小和

http://www.nowcoder.com/practice/edfe05a1d45c4ea89101d936cac32469

n = int(input())
nums = list(map(int, input().split()))
def merge_sort(left, right):
    if left >= right:
        return 0
    m = (left + right) // 2
    res = merge_sort(left, m) + merge_sort(m + 1, right)
    i, j = left, m + 1
    temp = [0] * (right - left + 1)
    index = 0
    while i <= m and j <= right:
        if nums[i] <= nums[j]:
            res += nums[i] * (right - j + 1)
            temp[index] = nums[i]
            index += 1
            i += 1
        else:
            temp[index] = nums[j]
            j += 1
            index += 1
    while j < right + 1 or i < m + 1:
        if i < m + 1:
            temp[index] = nums[i] 
            i += 1
        else:
            temp[index] = nums[j]
            j += 1
        index += 1
    nums[left:right + 1] = temp
        
    return res
print(merge_sort(0, len(nums) - 1))
全部评论

相关推荐

dao_yi:投了1000个左右,回消息的很少,要简历然后说过几天联系的都没有消息了,约面试的基本都是3000左右,足够在当地生活,最后去了一个武汉的3000,干了两天回来考研了,感觉这个行业加班是常态,看能不能考研上岸找个国企,或者大厂。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务