Darle una variedad de números. La fórmula de cálculo del arreglo "prefijo suma" es: rtSum [i] = sum (nums [0]… nums [i]). Devuelve la suma de números del prefijo
Ejemplo:
Entrada: nums = [1,2,3,4]
Salida: [1,3,6,10]
Explicación: El prefijo y el proceso de cálculo son [1, 1 + 2, 1 + 2 + 3, 1 + 2 + 3 + 4].
Solución 1: se define una matriz adicional para almacenar la suma del prefijo de cada elemento
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]);
}
}
La complejidad temporal de este algoritmo es O (n), la complejidad espacial es O (n) y la complejidad espacial es demasiado grande.
Solución 2: Utilice la matriz original para guardar el prefijo y
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;
}
La complejidad temporal del algoritmo es O (n) y la complejidad espacial es O (1)