AcWing 797. 差分
输入一个长度为 n 的整数序列。
接下来输入 m 个操作,每个操作包含三个整数 l,r,c,表示将序列中 [l,r] 之间的每个数加上 c。
请你输出进行完所有操作后的序列。
输入格式
第一行包含两个整数 n 和 m。
第二行包含 n 个整数,表示整数序列。
接下来 m 行,每行包含三个整数 l,r,c,表示一个操作。
输出格式
共一行,包含 n 个整数,表示最终序列。
数据范围
1≤n,m≤100000,
1≤l≤r≤n,
−1000≤c≤1000,
−1000≤整数序列中元素的值≤1000
输入样例:
6 3
1 2 2 1 2 1
1 3 1
3 5 1
1 6 1
输出样例:
3 4 5 3 4 2
题意:对序列进行区域内全部元素增值或减值
给定数组b,再构建b的前缀和数组a,那么b就称为a的差分数组,a称为b的前缀和数组
差分相当于是前缀和的一种逆运算,因此构造方法可以类似的写出,逆写方程通过前缀和数组a来构造差分数组b
for(int i=1;i<=n;i++) {
a[i]=tab.nextInt();
}
for(int i=n;i>0;i--) {
b[i]=a[i]-a[i-1];
}
差分的应用主要在于对区间整体进行修改,如将数组a[ ]的l,r区间内的值+c时,根据差分的定义,只需构建a的差分数组b,将b[ l ]+c,且b[ r+1 ]-c,这样即可以将l坐标后的值加c,同时由于+c-c,r坐标后的值不受影响
同理也可使用这个方法构建差分数组,定义insert函数来表示在差分数组中修改一个值
public static void insert(int l,int r,int c) {
b[l]+=c;
b[r+1]-=c;
}
在差分数组初始化为0的情况下,向区域范围为1的差分数组中插入一个a[i]值即可构建差分数组
for(int i=1;i<=n;i++) {
insert(i,i,a[i]);
}
代码如下
import java.io.*;
import java.util.*;
import javax.swing.text.DefaultEditorKit.InsertBreakAction;
public class Main {
static Scanner tab = new Scanner(System.in);
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
static int N = 1000010;
static int n,m;
static int a[]=new int [N];
static int b[]=new int [N];
public static void insert(int l,int r,int c) {
b[l]+=c;
b[r+1]-=c;
}
public static void main(String[] args) throws IOException {
n=tab.nextInt();
m=tab.nextInt();
for(int i=1;i<=n;i++) {
a[i]=tab.nextInt();
}
for(int i=1;i<=n;i++) {
insert(i,i,a[i]);
}
while(m-->0) {
int l=tab.nextInt();
int r=tab.nextInt();
int c=tab.nextInt();
insert(l, r, c);
}
for(int i=1;i<=n;i++) {
a[i]=a[i-1]+b[i];
System.out.print(a[i]+" ");
}
}
}
或是常规构建方法,简化使用一个数组滚动跟新即可
import java.io.*;
import java.util.*;
import javax.swing.text.DefaultEditorKit.InsertBreakAction;
public class Main {
static Scanner tab = new Scanner(System.in);
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
static int N = 1000010;
static int n,m;
static int a[]=new int [N];
static int b[]=new int [N];
public static void main(String[] args) throws IOException {
n=tab.nextInt();
m=tab.nextInt();
for(int i=1;i<=n;i++) {
a[i]=tab.nextInt();
}
for(int i=n;i>=1;i--) {
a[i]=a[i]-a[i-1];
}
while(m-->0) {
int l=tab.nextInt();
int r=tab.nextInt();
int c=tab.nextInt();
a[l]+=c;
a[r+1]-=c;
}
for(int i=1;i<=n;i++) {
a[i]+=a[i-1];
System.out.print(a[i]+" ");
}
}
}