前缀和:
一维:
#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
typedef long long ll;
const int N=1e6+10;
int a[N],s[N];
int main()
{
IOS
int n,m;cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>a[i];
s[i]=s[i-1]+a[i];//前缀和
}
while(m--)
{
int a,b;cin>>a>>b;
cout<<s[b]-s[a-1]<<endl;
}
return 0;
}
二维:
#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
typedef long long ll;
const int N=1e6+10;
int a[1010][1010];
int s[1010][1010];
signed main()
{
IOS
int n,m,q;cin>>n>>m>>q;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
cin>>a[i][j];
s[i][j]=a[i][j]+s[i-1][j]+s[i][j-1]-s[i-1][j-1];
}
}
while(q--)
{
int x1,y1,x2,y2;
cin>>x1>>y1>>x2>>y2;
cout<<s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]<<endl;
}
return 0;
}
差分:
一维:
#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
typedef long long ll;
const int N=1e6+10;
int a[N],s[N];
int n,m;
void insert(int l,int r,int c)
{
s[l]+=c,s[r+1]-=c;
}
signed main()
{
IOS
int q;cin>>n>>q;
for(int i=1;i<=n;i++)
{
cin>>a[i];
insert(i,i,a[i]);
}
while(q--)
{
int l,r,c;cin>>l>>r>>c;
insert(l,r,c);
}
for(int i=1;i<=n;i++) s[i]+=s[i-1];
for(int i=1;i<=n;i++) cout<<s[i]<<' ';
return 0;
}
二维:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=1e3+5;
map<ll,ll> mp;
int d[N];
ll yu[N][N],gz[N][N];
void ins(int x1,int y1,int x2,int y2,int c)
{
gz[x1][y1]+=c;
gz[x1][y2+1]-=c;
gz[x2+1][y1]-=c;
gz[x2+1][y2+1]+=c;
}
int main()
{
IOS
int n, m ,q;cin>>n>>m>>q;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
cin>>yu[i][j];
ins(i,j,i,j,yu[i][j]);
}
}
while(q--)
{
int a,b,c,d,v;cin>>a>>b>>c>>d>>v;
ins(a,b,c,d,v);
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
gz[i][j]+=gz[i-1][j]+gz[i][j-1]-gz[i-1][j-1];
}
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
cout<<gz[i][j]<<' ';
}cout<<endl;
}
return 0;
}