算法 一维数组前缀和

给你一个数组 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)

你可能感兴趣的:(算法)