题目
假定有一个无限长的数轴,数轴上每个坐标上的数都是 0。
现在,我们首先进行 n 次操作,每次操作将某一位置 x 上的数加 c。
接下来,进行 m 次询问,每个询问包含两个整数 l 和 r,你需要求出在区间 [l,r] 之间的所有数的和。
输入
数据的次数和查询的次数
3 3
每次进行下标位置的加减
1 2
3 6
7 5
查询left到right区间内的和
1 3
4 6
7 8
输出
8
0
5
public class 离散化_区间和{
public static void main(String[] args) {
//接收n和m,创建a和s数组,创建alls的arraylist,创建hashmap分别为add和query
Scanner in = new Scanner(System.in);
int n = in.nextInt(), m = in.nextInt(), N = 300010;
int[] a = new int[N], s = new int[N];
ArrayList<Integer> alls = new ArrayList<>();
注意一下,这里必须用自己创建的数据格式,原因在下面link类那里我有讲
ArrayList<link> add = new ArrayList<>();
ArrayList<link> query = new ArrayList<>();
//add导入数据
for (int i = 0; i<n; i++) {
int x = in.nextInt(), c = in.nextInt();
add.add(new link(x, c));
alls.add(x);
}
//query导入数据
for (int i = 0; i<m; i++) {
int l = in.nextInt(), r = in.nextInt();
query.add(new link(l, r));
alls.add(l);
alls.add(r);
}
//对alls进行去重并且排序
alls = new ArrayList<>(new HashSet<>(alls));
Collections.sort(alls);
//进行离散化
for(link item : add) {
a[find(alls, item.Key)] += item.Value;
}
//求出a的前缀和,用s接收, 切记这里是i=1,并且i<=,因为前缀和是从1开始的,0是空的
for (int i = 1; i<=alls.size(); i++) {
s[i] += s[i-1] + a[i];
}
//最后,根据每次的查询得出区间和
for (link item : query) {
int l = find(alls, item.Key), r = find(alls, item.Value);
System.out.println(s[r] - s[l-1]);
}
}
//二分查找
public static int find(ArrayList<Integer> index_list, int value){
int l = 0, r = index_list.size()-1;
while (l < r) {
int mid = (l+r)/2;
if (index_list.get(mid)>=value) {r = mid;}
else {l = mid + 1;}
}
return r + 1; //因为数组第一位为0
}
}
//这里创建一个key和value的数据形式,为什么不使用map是因为key会有重复
//例如:-1674和812是一对,但后面传入-1674和211就会覆盖掉之前的812,就只能自己写个数据形式了
class link {
public int Key;
public int Value;
public link (int key, int value){
this.Key = key;
this.Value = value;
}
}
附上y总的图
这是学算法以来耗时最长的一次,课听了1个小时,理解一个小时,自己敲三遍一个小时,啧,还是我太笨了,y总讲挺好的
声明:算法思路来源为y总,详细请见https://www.acwing.com/
本文仅用作学习记录和交流