差值维护和前缀和维护是一对孪生姐妹,只不过一个数组装的是前n项的和,而这个是n与n-1的差。
如题:给定n个数ai,有m次操作,每个操作是给al~ar增加一个数k。最终输出操作完后的这n个数的值。
要求:一个O(n+m)的做法。
for (i=1; i<=n; i++) scanf("%d",&b[i]);
for (i=1; i<=m; i++)
{
scanf("%d%d%d",&L,&R,&K);
s[L]+=K; s[R+1]-=K;
}
for (i=1; i<=n; i++) a[i]=a[i-1]+s[i];
for (i=1; i<=n; i++) printf("%d ",a[i]+b[i]);
以下便是其思路:
令s[i]=a[i]-a[i-1]。则a[i]=s[1]+s[2]+…+s[i]=a[i-1]+s[i]。
对于每次操作,相当于是令s[l]+=k,s[r+1]-=k。
同理,矩阵差值如下:给定一个n*n的矩阵ai,j,有m次操作,每个操作是给一个矩阵增加一个数k。最终输出操作完后的整个矩阵的值。
要求:一个O(n*n+m)的做法。
for (i=1; i<=n; i++)
for (j=1; j<=n; j++)
a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+s[i][j];
差值维护与前缀和类似,实用简单方便。