LeetCode 370. Range Addition （范围加法）$

Posted 2020-10-07 几米空间

Assume you have an array of length n initialized with all 0\'s and are given k update operations.

Each operation is represented as a triplet: [startIndex, endIndex, inc] which increments each element of subarray A[startIndex ... endIndex] (startIndex and endIndex inclusive) with inc.

Return the modified array after all k operations were executed.

Example:

Given:

    length = 5,
    updates = [
        [1,  3,  2],
        [2,  4,  3],
        [0,  2, -2]
    ]

Output:

    [-2, 0, 3, 5, 3]

 

Explanation:

Initial state:
[ 0, 0, 0, 0, 0 ]

After applying operation [1, 3, 2]:
[ 0, 2, 2, 2, 0 ]

After applying operation [2, 4, 3]:
[ 0, 2, 5, 5, 3 ]

After applying operation [0, 2, -2]:
[-2, 0, 3, 5, 3 ]

 

---

 

题目标签：Array　

　　题目给了我们一个2d updates， 和一个 length， 让我们返回一个 size = length 的array， 是经过 updates 的范围加法改动过的。

　　因为题目hint 说了要时间复杂度O(k+n)。所以我们不能遇到每一个update，都去array 里改动一次。

　　先遍历updates， 对于每一个update，我们只需要 标记 范围开始的 的那个number 和 范围结束的那个 number 的后一个。这里相当于，给每一个update 都规定了一个范围，开始是加，结尾后一个数字是减。

　　再遍历res array，设一个sum = 0， 对于每一个number， 把number 的值加入sum里， 再把number = sum。这里的意思就是遇到任何一个范围开始的时候，进行累加，因为我们只更改了开头和结尾，所以中间都是没更改过的值，或者是其他的范围开头结尾。累加的作用就是把中间没改过的number 都补上该有的值。

　　举例来看一下：

　　updates = [1,3,2] 和 [2,4,3]，length = 5

　　0 0 0 0 0

　　先遍历updates， 把开头和结尾标记

　　0 2 0 0 -2　　index 1 = 2；index 3+1 = -2；

　　0 2 3 0 -2　　index 2  = 3；index 4+1 超出了范围，就不用处理。

　　遍历res array，进行累加 sum += res[i], res[i] = sum

　　0 2 3 0 -2　　sum = 0+0

　　0 2 3 0 -2　　sum = 0+2

　　0 2 5 0 -2　　sum = 2+3

　　0 2 5 5 -2　　sum = 5+0

　　0 2 5 5 3　　 sum = 5-2　　

　　可以看到，从第一个范围开头开始，sum 进行累加，并更新number，如果遇到另一个范围，继续累加，如果遇到任何一个范围结束，把那一个范围累加的值减去，这个范围的加法就结束了，继续其他的。

 

Java Solution:

Runtime beats 77.60% 

完成日期：09/16/2017

关键词：Array

关键点：只需要标记范围开始，和结束的位置，之后进行累加

 1 class Solution 
 2 {
 3     public int[] getModifiedArray(int length, int[][] updates) 
 4     {
 5         int [] res = new int[length];
 6         
 7         // iterate each operation
 8         for(int[] update: updates)
 9         {
10             int start = update[0];
11             int end = update[1];
12             int val = update[2];
13             
14             // mark first element
15             res[start] += val;     
16             // mark last element (end + 1)
17             if(end + 1 < length)
18                 res[end + 1] -= val;
19         }
20         
21         int sum = 0;
22         for(int i=0; i<length; i++)
23         {
24             sum += res[i];
25             res[i] = sum;
26         }
27         
28         return res;
29     }
30 }

参考资料：

https://discuss.leetcode.com/topic/49691/java-o-k-n-time-complexity-solution

 

LeetCode 题目列表 - LeetCode Questions List