蓝桥杯题目-肖恩的投球游戏加强版

链接

肖恩的投球游戏加强版 - 蓝桥云课 (lanqiao.cn)

知识点

二维前缀和，二维差分

思路

       定义三个数组，分别为原数组a，二维差分数组diff，二维前缀和数组prefix，首先对原数组求差分数组diff，在通过差分数组实现以(x1,y1)为左上角，(x2,y2)为右上角的矩阵都加c，然后再通过对差分数组求前缀和，得到前缀和数组prefix，再把前缀和数组打印输出即可。

代码

#include<bits/stdc++.h>
using namespace std;

const int N = 1010;
int a[N][N];   //原数组
int diff[N][N];  //差分数组
int prefix[N][N];  //前缀和数组

int main(){
  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];
    }
  }

  //对原数组做二维差分
  for(int i=1;i<=n;i++){
    for(int j=1;j<=m;j++){
      diff[i][j] = a[i][j] - a[i-1][j] -a[i][j-1] + a[i-1][j-1];
    }
  }

  //以(x1,y1)为左上角，(x2,y2)为右上角的球框矩阵都投入c个球
  while(q--){
    int x1,y1,x2,y2,c;
    cin>>x1>>y1>>x2>>y2>>c;
    diff[x1][y1] += c;
    diff[x1][y2+1] -= c;
    diff[x2+1][y1] -= c;
    diff[x2+1][y2+1] += c;
  }
  
  //对二维差分数组做前缀和
  for(int i=1;i<=n;i++){
    for(int j=1;j<=m;j++){
      prefix[i][j] = prefix[i-1][j] + prefix[i][j-1] -prefix[i-1][j-1] + diff[i][j];
    }
  }

   for(int i=1;i<=n;i++){
    for(int j=1;j<=m;j++){
      cout<<prefix[i][j]<<" ";
    }
    cout<<endl;
  }
  return 0;
}