JAVA文件如何差分,AcWing 797. 差分_java

什么是差分

差分是求前缀和的逆操作，对于原数组a[n],构造出一个数组b[n],使a[n]为b[n]的前缀和。一般用于快速对整个数组进行操作，比如对将a数组中[l,r]部分的数据全部加上c。使用暴力方法的话，时间复杂至少为O(n)，而使用差分算法可以将时间复杂度降低到O(1)。

算法思路

拥有数组b[n]后，想要对a数组中所有的数据加上c,只需要将b[1]+c即可，因为a[i]是b[i]的前缀和，a[i]=b[1]+b[1]+b[3]+……+b[n]。b[1]是所有的a[i]都拥有的子元素,将b[0]+c,那么a[n]中所有的数据都会加上c。如果想将a数组中[l,r]部分的数据全部加上c，只需要将b[l]+c,然后b[r+1]-c即可。

差分操作和前缀和一样数组下标都从1开始。

b[l]+c后，l后面的数组都会加c。r后面的数据也会被改变，要改回来就得b[r+1]-c

如何构造b[n]

构造b[n]看起来很难，其实根本就不用刻意去构造它。

如果将a数组中[l,r]部分的数据全部加上c看作一次插入操作，那么构造的过程可以看作是将a进行了n次插入操作。第一次在[1,1]的范围插入了a[1],第二次在[2,2]范围插入a[2]，第二次在[3,3]范围插入a[3]……，以此类推，进行n次插入后，那么数组a就正好是数组b的前缀和了。

例题

原题链接

输入一个长度为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

java代码

package 差分;

import java.io.BufferedInputStream;

import java.util.Scanner;

public class Main {

static int N = 100010;

public static void main(String[] args) {

Scanner scanner = new Scanner(new BufferedInputStream(System.in));

int n = scanner.nextInt();

int m = scanner.nextInt();

//a为原数组，b为差分数组

int[] a = new int[N];

int[] b = new int[N];

for (int i = 1; i <= n; i++) {

a[i] = scanner.nextInt();

}

//进行n次插入，初始化差分数组

for(int i=1;i<=n;i++) {

insert(b, i, i, a[i]);

}

while(m-->0) {

int l,r,c;

l = scanner.nextInt();

r = scanner.nextInt();

c = scanner.nextInt();

insert(b, l, r, c);

}

//经过一系列插入操作后，现在答案数组应该是b数组的前缀和，让b数组变成b的前缀和。

//公式 b[i] = b[i-1] + b[i]

for(int i=1;i<=n;i++)b[i] +=b[i-1];

for(int i=1;i<=n;i++)System.out.print(b[i]+" ");

System.out.println();

scanner.close();

}

//插入操作函数

public static void insert(int[] a,int l,int r,int c) {

a[l]+=c;

a[r+1]-=c;

}

}