import java.util.Scanner;
public class XiaoEnBall3693 {
public static void main(String[] args) {
Scanner s = new Scanner(System.in);
int n = s.nextInt();
int q = s.nextInt();
int []a = new int [n+5];//每次多开几位,防止越界//
for(int i = 1; i <= n ; i ++){
a[i] = s.nextInt();
}//初始化数据数组时从1开始,因为最后前缀和递推会用到公式a[i] = a[i-1] + diff[i]//
int []diff = new int[n+5];
//数组多开几位,防止diff[r+1]越界//
for(int i = 1; i <= n; i ++){
diff[i] = a[i] - a[i-1];
}
//初始化diff放在记录区间变化前,同样从1开始,因为用到差分公式diff[i] = a[i] - a[i-1]//
for(int i = 0; i < q;i ++){
int l = s.nextInt();
int r = s.nextInt();
int c = s.nextInt();
diff[l] += c;
diff[r+1] -= c;
}
//数组1开,题中给出的查询l,r不用-1,直接用//
for(int i = 1; i <= n; i ++){
a[i] = a[i-1] + diff[i];
System.out.print(a[i]+" ");
}
}
}
被差分折磨的一天
不得不说,差分里面的索引真的容易把作为算法小白的鼠鼠给绕晕。捋了好几遍终于是弄清楚了。
差分总结
差分数组(diff)可以理解为前缀和数组的逆数组。它以记录原始数据数组前后两项差的形式,使得自己能通过前缀和相加的模式计算得到原数组。也就是说,差分数组的前缀和数组就是原始数据数组。
其主要作用在于,遇到区间(l,r)上m次同时增加或者减少x时,能通过对差分数组单点操作,即diff[l]+=x;diff[r+1]-=x,使得对它前缀和相加后,整体以O(n+m)的时间复杂度实现区间变化;而使用循环嵌套的暴力解法需要O(n*m)的时间复杂度。当n,m均较大时,使用差分可以大大降低时间复杂度。
差分模版
上面给出的代码块其实就是一个经典的差分模版。差分数组初始化后,第i项元素表示原数据数组第i项与第i-1项的差。根据(l,r),diff的第l项加x,第r+1项减x,前缀和相加(a[i] = a[i-1] + diff[i]),使得区间(l,r)上均增加x。
对我来说,写对差分代码的关键在于写对索引值。为了防止越界,我设计所有数组的初始化与相加都从1开始,到n结束。
例题分析
纯属于模版题,思路就是基本的差分运用。
强化练习
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner s = new Scanner(System.in);
int n = s.nextInt();
int m = s.nextInt();
int q = s.nextInt();
long [][]a = new long [n+5][m+5];
for(int i = 1; i <= n; i ++){
for(int j = 1; j<= m; j ++){
a[i][j] = s.nextInt();
}
}
long [][]diff = new long [n+5][m+5];
for(int i = 1;i <= n; i ++){
for(int j = 1; j <= m; j ++){
diff[i][j] = a[i][j]-a[i][j-1];
}
}
for(int i = 0; i < q;i ++){
int x1 = s.nextInt();
int y1 = s.nextInt();
int x2 = s.nextInt();
int y2 = s.nextInt();
int c = s.nextInt();
for(int j = x1; j <=x2; j ++){
diff[j][y1] +=c;
}
for(int z = x1; z <=x2; z ++){
diff[z][y2+1] -=c;
}
}
for(int i = 1; i <=n; i ++){
for(int j = 1; j <=m; j ++){
a[i][j] = a[i][j-1] + diff[i][j];
System.out.print(a[i][j]+" ");
}
System.out.println();
}
}
}
例题分析
在理解了前一道题目的情况下,这道加强题也只是多了一维数组。也就是对于每一行讨论第一题的情况。但多一维数组就多一维索引,鼠鼠总是在奇怪的地方出了岔子,要不就是漏了等号,要不就是把m敲成n,总之调了几次才过,以后类似的题目需要仔细检查for循环部分的索引初始值、结束值、等号。