什么是差分数组？

最新推荐文章于 2025-12-28 17:52:30 发布

原创最新推荐文章于 2025-12-28 17:52:30 发布 · 819 阅读

17 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

algorithms 同时被 2 个专栏收录

30 篇文章

订阅专栏

leetcode

3 篇文章

订阅专栏

文章介绍了差分数组的概念，如何通过差分数组快速计算数组元素的变化，以及在LeetCode例题中如何使用差分数组进行航班预订统计。展示了两种处理数组增减操作的方法：直接和通过差分数组优化。

定义

对于数组 A[n]，它的差分数组为：
$diff[i]=\begin{cases} A[i],&i==0 \\ A[i]-A[i-1],&0<i<n \end{cases}$
显然，通过差分数组 diff[n]，可以求得 A[n] 中的某一具体元素：
$A[i]=\begin{cases} diff[i],&i==0 \\ diff[i]+A[i-1],&0<i<n \end{cases}$

应用

数组 A[n] 从下标 j 开始的元素，都增加 v。(即 A[j, j+1, ..., n-1] += v)
$d i ff [j] = d i ff [j] + v$
数组 A[n] 在区间 [L, R] 内的元素，都增加 v。(即 A[L, L+1, ..., R] += v)
$\begin{cases} diff[L]=diff[L]+v \\ diff[R+1]=diff[R+1]-v \end{cases}$
数组 A[n] 中的某一元素 A[j] 增加 v。
$\begin{cases} diff[j] = diff[j]+v \\ diff[j+1]=diff[j+1]-v \end{cases}$

例题

LeetCode1109. 航班预订统计
暴力解

int* corpFlightBookings(int** bookings, int bookingsSize, int* bookingsColSize, int n, int* returnSize) {
    int *ret = (int*)calloc(n, sizeof(int));
    memset(ret, 0, n * sizeof(int));

    for (int i = 0; i < bookingsSize; i++)
    {
        for (int j = bookings[i][0] - 1; j < bookings[i][1]; j++)
        {
            ret[j] += bookings[i][2];
        }
    }
    *returnSize = n;

    return ret;
}

差分数组

int* corpFlightBookings(int** bookings, int bookingsSize, int* bookingsColSize, int n, int* returnSize) {
    int *ret = (int*)calloc(n, sizeof(int)), *diff = (int*)calloc(n, sizeof(int));
    memset(ret, 0, n * sizeof(int));
    memset(diff, 0, n * sizeof(int));
    *returnSize = n;

    for (int i = 0; i < bookingsSize; i++)
    {
        diff[bookings[i][0] - 1] += bookings[i][2];
        if (bookings[i][1] < n)
            diff[bookings[i][1]] -= bookings[i][2];
    }
    ret[0] = diff[0];
    for (int i = 1; i < n; i++)
        ret[i] = diff[i] + ret[i - 1];

    return ret;
}