鼠鼠的算法题解（差分）

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循环部分的索引初始值、结束值、等号。