给你一个数组 nums 。数组「前缀和」的计算公式为:rtSum[i] = sum(nums[0]…nums[i]) 。请返回 nums 的前缀和
示例 :
输入:nums = [1,2,3,4]
输出:[1,3,6,10]
解释:前缀和计算过程为 [1, 1+2, 1+2+3, 1+2+3+4] 。
解法一:额外定义一个数组,用于存放每一项的前缀和
int PreSum1(int *arr,int *Sum,int arr_length) //o(n) o(n)
{
if(arr_length <= 0)
return NULL;
for (int i = 0;i < arr_length;i++)
{
if (i == 0)
Sum[i] = arr[i];
else
Sum[i] = Sum[i - 1] + arr[i];
printf("%d ", Sum[i]);
}
}
该算法的时间复杂度为O(n),空间复杂度为O(n),其中空间复杂度太大.
解法二:利用原数组保存前缀和
int PreSum2(int* nums, int numsSize, int* returnSize) //O(n),O(1)
{
for(int i=1;i<numsSize;i++)
{
nums[i] += nums[i-1];
}
*returnSize = numsSize;
return nums;
}
该算法的时间复杂度为O(n),空间复杂度为O(1)