蓝桥杯大学C组真题--重新排列--python解

重新排序

问题描述

给定一个数组 A 和一些查询 ,L i ,R i , 求数组中第 L i至第 R i个元素之和。

小蓝觉得这个问题很无聊, 于是他想重新排列一下数组, 使得最终每个查 询结果的和尽可能地大。小蓝想知道相比原数组, 所有查询结果的总和最多可 以增加多少?

输入

输入第一行包含一个整数 n 。第二行包含 n 个整数 1,2,⋯,A 1 ,A 2 ,⋯,A n , 相邻两个整数之间用一个空格分隔。

第三行包含一个整数 m 表示查询的数目。接下来 m 行, 每行包含两个整数、L i 、R i, 相邻两个整数之间用一个空格分隔。

思路

利用到差分来计算数组间的重叠的关系，暴力只有60%的通过

60%代码

n = int(input())
lis = list(map(int,input().split()))
x = int(input())


dic = [0] * n
num1 = 0

for i in range(x):
    a,b = map(int,input().split())
    for j in range(a-1,b):
        num1 += lis[j]
        dic[j] += 1


dic = sorted(dic)
lis = sorted(lis)

num = 0
for i in range(n):
    num += dic[i] * lis[i]

print(num - num1)

AC参考代码

n = int(input())
lis = [0] + list(map(int,input().split())) + [0]      # 初始化

x = int(input())

dic = [0] * (n+2)   # 标记lr之间出现的次数，相交重叠什么的


mark = [0] * (n+2)

for i in range(x):
    a,b = map(int,input().split())      # 输入左右的范围
    mark[a] += 1
    mark[b+1] -= 1

for j in range(1,n+1):
    dic[j] += dic[j-1] + mark[j]        # 主要代码，求出相叠的，每个点在左右范围内出现的次数

num1 = 0
for i in range(1,n+1):      # 求出原列表的值
    num1 += dic[i] * lis[i]

di = sorted(dic[1:n+1])     # 只要将重叠次数最多的放上最大的数，后面的依次类推。就是解
li = sorted(lis[1:n+1])

num = 0
for i in range(n):
    num += di[i] * li[i]        # 新列表的值

print(num - num1)

输入样例

5
1 2 3 4 5
2
1 3
2 5

输出样例

4