AcWing 802. 区间和
原题链接
简单
作者:
Broken_
,
2024-09-26 18:03:57
,
所有人可见
,
阅读 1
java
import java.io.BufferedInputStream;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Scanner;
/**
* 区间和
*/
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(new BufferedInputStream(System.in));
int n = sc.nextInt(); //n次操作
int m = sc.nextInt(); //m次询问
int N = 300010; //因为需要将所有x,l,r存在数组中,这样就是n + 2m <= 300000
int[] a = new int[N]; //从1开始,需要通过x找到离散量,然后+1,
int[] s = new int[N]; //前缀和来做,所以需要从1开始记录a
List<Integer> alls = new ArrayList<>(); //将所有的使用到的数存在alls中,比如x,l,r
List<Pairs> add = new ArrayList<>(); //用来存n次操作
List<Pairs> query = new ArrayList<>(); //用来存m次询问
for (int i = 0; i < n; i++) {
int x = sc.nextInt(); // 读取下一个增量数值
int c = sc.nextInt(); // 读取增量值
add.add(new Pairs(x, c)); // 将增量操作存储到add列表
alls.add(x); // 将x值存储到alls,待后续处理
}
for (int i = 0; i < m; i++) {
int l = sc.nextInt(); // 读取查询的左边界
int r = sc.nextInt(); // 读取查询的右边界
query.add(new Pairs(l, r)); // 存储查询操作到query列表
alls.add(l); // 将左边界l加入alls
alls.add(r); // 将右边界r加入alls
}
//到此为止,alls中存好了所有会被用到的数轴上的点,开始进行离散化处理
Collections.sort(alls);//排序
int unique = unique(alls);//去重
alls = alls.subList(0, unique); //去重后的数值只保留有效部分
for (Pairs item : add) {
int index = find(item.first, alls); // 找到x在离散化数组中的位置
a[index] += item.second; // 更新相应位置的增量值
}
//前缀和
for (int i = 1; i <= alls.size(); i++) s[i] = s[i - 1] + a[i];
for (Pairs item : query) {
int l = find(item.first, alls); // 找到查询左边界的位置
int r = find(item.second, alls); // 找到查询右边界的位置
System.out.println(s[r] - s[l - 1]); // 输出区间和
}
}
//去重
static int unique(List<Integer> list) {
int res = 0;
for (int i = 0; i < list.size(); i++) {
if (i == 0 || list.get(i) != list.get(i - 1)) {
list.set(res, list.get(i));
res++;
}
}
return res;
}
//二分
static int find(int x, List<Integer> list) {
int l = 0;
int r = list.size() - 1;
while (l < r) {
int mid = l + r >> 1;
if (list.get(mid) >= x) {
r = mid;
} else {
l = mid + 1;
}
}
return l + 1; //因为要考虑到前缀和
}
}
class Pairs {
int first;
int second;
public Pairs(int first, int second) {
this.first = first;
this.second = second;
}
}