题目
输入一个长度为 n 的整数序列。
接下来输入 m 个操作,每个操作包含三个整数 l,r,c表示将序列中 [l,r] 之间的每个数加上c。
请你输出进行完所有操作后的序列。
输入
两个初始值,分别为元素个数,以及进行计算的次数
6 3
数组
1 2 2 1 2 1
计算,每次三个数据,分别为起始,终点,以及加减的数
1 3 1
3 5 1
1 6 1
输出
3 4 5 3 4 2
public class 一维差分 {
public static void main(String[] args){
//初始化数据
Scanner in = new Scanner(System.in);
int n = in.nextInt(), times = in.nextInt();
int[] a = new int[n+2], b = new int[n+2];
//导入原数组
for (int i = 1; i<=n; i++) {a[i] = in.nextInt();}
//计算差分数组,核心思想为前缀和逆运算,也就是b[i] = a[i]-a[i-1]
for (int i = 1; i<=n; i++) {insert(b, i, i, a[i]);}
//对差分数组某区域整体加减,核心思想为,l-r下标范围整体+c,那么就将(l)+c,(r+1)-c
//这样在计算前缀和的时候达到自动进行全部加减,只影响区域内的数据
while (times-- > 0) {
int l = in.nextInt(), r = in.nextInt(), c = in.nextInt();
insert(b, l, r, c);
}
//将b数组转为b的前缀和数组
for (int i = 1; i<=n; i++) {b[i]+=b[i-1];}
//输出结果
for (int i = 1; i<=n; i++) {
System.out.print(b[i]+" ");
if (i==n) System.out.println();
}
}
//insert,差分的模板运算
//1.使用在差分数组初始化时
//2.使用在进行对差分数组某区域整体加减时
public static void insert(int[] b, int l, int r, int c){
b[l] += c;
b[r+1] -= c;
}
}
声明:算法思路来源为y总,详细请见https://www.acwing.com/
本文仅用作记录和交流