一维前缀和
1、什么是前缀和
例如:给你一个数组a[5]={1,3,4,6,7};注意:(让它的下标从1开始),求这个数组的前缀和的过程是这样的。
sum[0]=0;
sum[1]=sum[0]+a[1]=a[1]=1;
sum[2]=sum[1]+a[2]=a[1]+a[2]=4;
sum[3]=sum[2]+a[3]=a[1]+a[2]+a[3]=8;
sum[4]=sum[3]+a[4]=a[1]+a[2]+a[3]+a[4]=14;
sum[5]=sum[4]+a[5]=a[1]+a[2]+a[3]+a[4]+a[5]=21;
所以数组a[5]={1,3,4,6,7};的前缀和数组就是这样的sum[6]={0,1,4,8,14,21};
sum[0] = 0;
for (i=1; i<=n; i++)
{
cin >> a[i];
sum[i] = sum[i-1] + a[i];//核心
}
如果你看得不得劲儿还可以这样
还是那个数组a[5]={1,3,4,6,7};注意:这时我让它的下标从0开始。
sum[0]=0;
sum[1]=sum[0]+a[0]=a[0]=1;
sum[2]=sum[1]+a[1]=a[0]+a[1]=4;
sum[3]=sum[2]+a[2]=a[0]+a[1]+a[2]=8;
sum[4]=sum[3]+a[3]=a[0]+a[1]+a[2]+a[3]=14;
sum[5]=sum[4]+a[4]=a[0]+a[1]+a[2]+a[3]+a[4]=21;
sum[0]=0;
for (i=0; i<n; i++)
{
cin >> a[i];
sum[i+1] = sum[i] + a[i];//核心
}
另外还可以用前缀和来计算任意连续子串和。
#include <iostream>
#include <cstdio>
using namespace std;
int main()
{
int n, a[5555], i, sum[5555], j, f[5555];
int k = 0;
cin >> n;
sum[0] = 0;
for (i = 1; i <= n; i++)
{
cin >> a[i];
sum[i] = sum[i - 1] + a[i];
}
for (i = 0; i <= n; i++)
for (j = i + 1; j <= n; j++)
{
f[k++] = sum[j] - sum[i];
}
for (i = 0; i < k; i++)
cout << f[i] << " ";
return 0;
}
输入:5
1 3 4 6 7
输出:1 4 8 14 21 3 7 13 20 4 10 17 6 13 7
我觉得前缀和是解决一些问题的小技巧,学会了这个技巧很有用。