c++的区间修改
题目描述:
有一个 N× M 的矩阵 A, 操作 add(x1,y1,x2,y2 , k)表示对矩阵 A 的(x1,y1) 到(x2,y2)区域内的每个 数都加上 k。 有 P 个 add 操作, 输出 P 个 add 操作后的矩阵 A。
输入格式:
第一行 3 个正整数: N、 M、 P, 范围[1,1000]。
接下来, N 行 M 列描述矩阵 A,每个数的范围为[1,100];
后面紧跟 P 行, 每行五个整数, 分别为 x1,y1,x2,y2,k;
1<=x1<=x2<=N, 1<=y1<=y2<=M, k 范围[1,1000]
输出格式:
N 行 M 列的矩阵, 为 add 操作后的新矩阵。
样例输入:
3 3 2
1 2 3
4 5 6
7 8 9
2 1 3 2 4
1 2 2 3 1
样例输出:
1 3 4
8 10 7
11 12 9
提示:
用差分 + 前缀和进行降维, 时间复杂度变为 O(N*M+P), 1 秒内可轻松出解。
我自己写了个代码,,得分却一直是20分,,不知道问题出在哪里了
代码片
// An highlighted block
#include<bits/stdc++.h>
using namespace std;
int c[1500][1500];
int a[1500][1500];
void xiugai(int x1,int y1,int x2,int y2,int k){
c[x1][y1]+=k;
c[x1][y2+1]-=k;
c[x2+1][y1]-=k;
c[x2+1][y2+1]+=k;
return ;
}
int main()
{
int N,M,P;
cin>>N>>M>>P;
for(int i=1;i<=N;i++)
for(int j=1;j<=M;j++)
cin>>a[i][j];//输入数组
int x1,x2,y1,y2,k;
for(int i=1;i<=P;i++){//对辅助数组进行修改
cin>>x1>>y1>>x2>>y2>>k;
xiugai(x1,y1,x2,y2,k);
}
for(int i=1;i<=N;i++){
for(int j=1;j<=M;j++){
c[i][j]=c[i][j-1]+c[i-1][j]-c[i-1][j-1]+c[i][j];
a[i][j]+=c[i][j];//求前缀和
cout<<a[i][j]<<" ";
}
printf("\n");
}
return 0;
}
二维差分算法提示:
和一维类似。用数组 c 存储总改变量。在 c[x1][y1]处加上 a,在 c[x2+1][y1]和c[x1][y2+1]处减 a,在 c[x2+1][y2+1]再加上 a。最后(i,k)位置上的数值就是 c 数组在(i,k)位置的前缀和
有没有大神帮忙看一看!!!