acwing 802 区间和_离散化 Python注释版 蓝桥杯Python

离散化

• 这里特指整数离散化，且有序
• 思想：
  * a[]中可能有重复元素——去重
  * 算出x离散化后的值——二分
  * 求前缀和
  

'''
假定有一个无限长的数轴，数轴上每个坐标上的数都是 0。
现在，首先进行 n 次操作，每次操作将某一位置 x 上的数加 c。
接下来，进行 m 次询问，每个询问包含两个整数 l 和 r，需要求出在区间 [l,r] 之间的所有数的和。

输入格式
第一行包含两个整数 n 和 m。
接下来 n 行，每行包含两个整数 x 和 c。
再接下来 m 行，每行包含两个整数 l 和 r。

输出格式
共 m 行，每行输出一个询问中所求的区间内数字和。

数据范围
−10^9≤x≤10^9,
1≤n,m≤10^5,
−10^9≤l≤r≤10^9,
−10000≤c≤10000
输入样例：
3 3
1 2
3 6
7 5
1 3
4 6
7 8
输出样例：
8
0
5
'''

n, m = map(int, input().split())
# a数组用于存放离散数据的值，将原离散数组映射到数组a中，s是前缀和数组
a, s = [0] * 300010, [0] * 300010  # x,l,r 范围都是10^5
alls, add, query = [], [], []  # alls 存放a数组中离散数据的坐标，add存放添加数据的坐标+值，query存放询问区间的左右数值，用于后面遍历

for _ in range(n):
    x, c = map(int, input().split())
    add.append((x, c))
    alls.append(x)

for _ in range(m):
    l, r = map(int, input().split())
    query.append((l, r))
    alls.append(l)
    alls.append(r)

# 思想： 对alls进行去重并且排序
alls = list(set(alls))  # 集合不允许重复
alls.sort()  # 排序 sort是升序排列


# 用二分法找到数据的坐标
def find(d):
    l, r = 0, len(alls) - 1
    while l < r:
        mid = (l + r) // 2
        if alls[mid] >= d:
            r = mid
        else:
            l = mid + 1
    return l + 1  # 也可以返回l l+1是将数据离散化成1,2,3,4...n(从1开始，而不是从0开始) 计算前缀和更方便


# 将所有的数插入a数组
for x, c in add:
    index = find(x)
    a[index] += c

# 处理前缀和
for i in range(1, 300010):
    s[i] = s[i - 1] + a[i]

for l,r in query:
    print(s[find(r)]-s[find(l)-1])