重新排序线段树python

试题 F: 重新排序 时间限制: 1.0s 内存限制: 512.0MB 本题总分：15 分

【问题描述】 给定一个数组 A 和一些查询 Li , Ri，求数组中第 Li 至第 Ri 个元素之和。 小蓝觉得这个问题很无聊，于是他想重新排列一下数组，使得最终每个查 询结果的和尽可能地大。小蓝想知道相比原数组，所有查询结果的总和最多可 以增加多少?

【输入格式】 输入第一行包含一个整数 n。 第二行包含 n 个整数 A1, A2, · · · , An，相邻两个整数之间用一个空格分隔。 第三行包含一个整数 m 表示查询的数目。 接下来 m 行，每行包含两个整数 Li、Ri ，相邻两个整数之间用一个空格分 隔。

【输出格式】 输出一行包含一个整数表示答案。

【样例输入】

5

1 2 3 4 5

2

1 3

2 5

【样例输出】 4

【样例说明】 原来的和为 6 + 14 = 20，重新排列为 (1, 4, 5, 2, 3) 后和为 10 + 14 = 24，增 加了 4。

【评测用例规模与约定】 对于 30% 的评测用例，n, m ≤ 50 ； 对于 50% 的评测用例，n, m ≤ 500 ； 对于 70% 的评测用例，n, m ≤ 5000 ； 对于所有评测用例，1 ≤ n, m ≤ 105，1 ≤ Ai ≤ 106，1 ≤ Li ≤ Ri ≤ 106 。

不知道有什么bug，也没测评数据，过了80，就是写了个板子

n=int(input())
num=[0]+list(map(int,input().split()))
class node():
    def __init__(self,q=0):

        self.lazy=q



tree=[node() for i in range(n*4+100)]


def pushdown(x,l,r):
    if (l==r):
        return
    if (tree[x].lazy):

        tree[x<<1].lazy+=tree[x].lazy
        tree[x<<1|1].lazy += tree[x].lazy
        tree[x].lazy=0
    return

def update(x,l,r,a,b):
    if (a<=l and b>=r):
        tree[x].lazy+=1

        return
    #print(x,l,r)
    pushdown(x,l,r)
    mid=(l+r)>>1
    if (a<=mid):
        update(x<<1,l,mid,a,b)
    if (b>mid):
        update(x<<1|1,mid+1,r,a,b)
nums=[0 for i in range(n+1)]
def quary(x,l,r):
    #print(x,l,r,tree[x].lazy)
    if (l==r):
        nums[l]=tree[x].lazy

        return
    pushdown(x,l,r)
    mid=(l+r)>>1
    quary(x<<1,l,mid)
    quary(x<<1|1,mid+1,r)
    return
m=int(input())
for _ in range(m):
    a,b=map(int,input().split())
    if (a>n):
        continue
    elif b>n:
        b=n

    #print(a,b)
    update(1,1,n,a,b)
quary(1,1,n)
#print(nums)
yuanben=0
del tree
for i in range(1,n+1):
    yuanben+=nums[i]*num[i]
nums.sort()
num.sort()
xianzai=0
for i in range(1,n+1):
    xianzai+=nums[i]*num[i]
print(xianzai-yuanben)


"""
5
1 2 3 4 5
2
1 3
2 5
"""