【LeetCode(Java) - 370】区间加法

1、题目描述

在这里插入图片描述

2、解题思路

  创建一个 int[] 数组 ans,长度为 length。

  对于每一个给定的 [startIndex, endIndex, inc] 我们可以理解成如下:

  把 ans 的 [startIndex,length-1] 都加上了 inc,然后再把 [endIndex+1, length-1] 再减去 inc。

  具体的操作是:

  1、先 ans[startIndex] += val,ans[endIndex+1] += -val;

  3、然后 [startIndex,length-1] 遍历进行 ans[i] += ans[i-1]。

  这样做的目的是因为所有的三元组 [startIndex, endIndex, inc] 对数组 ans 的操作是独立的,我们先对所有的三元组对 ans 的操作在边界上做好,然后遍历一遍 ans 数组即可同时执行完所有三元组对 ans 的操作。

3、解题代码

class Solution {
    
    
   public int[] getModifiedArray(int length, int[][] updates) {
    
    
        int[] ans = new int[length];
        int start, end, val;
        for (int[] update : updates) {
    
    
            start = update[0];
            end = update[1];
            val = update[2];
            ans[start] += val;
            if (end < length - 1) {
    
    
                ans[end + 1] -= val;
            }
        }
        for (int i = 1; i < length; i++) {
    
    
            ans[i] += ans[i - 1];
        }
        return ans;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_29051413/article/details/108277476