LeetCode 370. 区间加法(Range Addition)

假设你有一个长度为 n 的数组,初始情况下所有的数字均为 0,你将会被给出 k 个更新的操作。

其中,每个操作会被表示为一个三元组:[startIndex, endIndex, inc],你需要将子数组 A[startIndex ... endIndex](包括 startIndex 和 endIndex)增加 inc

请你返回 k 次操作后的数组。

示例:
输入: length = 5, updates = [[1,3,2],[2,4,3],[0,2,-2]]
输出: [-2,0,3,5,3]

解释:
初始状态:
[0,0,0,0,0]
进行了操作 [1,3,2] 后的状态:
[0,2,2,2,0]
进行了操作 [2,4,3] 后的状态:
[0,2,5,5,3]
进行了操作 [0,2,-2] 后的状态:
[-2,0,3,5,3]

来源:力扣(LeetCode)

链接:力扣

著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请fangfa 

方法一(没会力扣会员用Vs初步测试通过):

思想:前缀和

public partial class HleetCode
{
    public static class N370RangeAdditon
    {
        public static int[] Commit(int length, int[][] updates)
        {
            //定义结果数组
            var res = new int[length];

            //定义前缀和数组
            var prefixSum = new int[length];

            for (int i = 0; i < updates.Length; i++)
            {
                int startIdex = updates[i][0];
                int endIndex = updates[i][1];
                int inc = updates[i][2];

                for (int j = 0; j < length; j++)
                {
                    //在区间左边的,前缀和数组不增加
                    if (j < startIdex) continue;

                    //在区间右边的,前缀和加上区间内inc的区间长度的倍数
                    if (j > endIndex)
                    {
                        prefixSum[j] += (endIndex - startIdex + 1) * inc;
                    }
                    //在区间内的,按照与端点的距离(区间长度)加inc的倍数
                    else
                    {
                        prefixSum[j] += (j - startIdex + 1) * inc;
                    }
                }
            }

            //将前缀和数组转化为原始数组
            res = PrefixSumArrayToOriginalArray(prefixSum);
            return res;
        }

         //将前缀和数组转化为原始数组
        public static int[] PrefixSumArrayToOriginalArray(int[] prefixSumArray)
        {
            for (int i = prefixSumArray.Length - 1; i > 0; i--)
            {
                prefixSumArray[i] = prefixSumArray[i] - prefixSumArray[i - 1];
            }
            return prefixSumArray;
        }
    }
}

 

上一篇:Unity 图文排版、超链接


下一篇:iOS简单的插入排序算法浅析