二维前缀和 codeforces 832C

#include <iostream>  #include <cstdio>  #include <cstring >  #include <iostream>  using namespace std;  int mp[102][102][11];  int sum[102][102][11];  int main()  {      int n,q,c;      while(~scanf("%d%d%d",&n,&q,&c)) {          memset(mp,0,sizeof(mp));          for(int i = 0; i < n; i++) {              int x,y,s;              scanf("%d%d%d",&x,&y,&s);              mp[x][y][s]++;       //记录地图 在（x，y）上 亮度为s的星星的个数  这里因为星星可以重叠在一个点 要用++          }          memset(sum,0,sizeof(sum));          for(int k = 0; k <= c; k++) {              for(int i = 1; i <= 100; i++) {       //二维前缀和                  for(int j = 1; j <= 100; j++) {                      sum[i][j][k] = sum[i-1][j][k]+sum[i][j-1][k]-sum[i-1][j-1][k]+mp[i][j][k];      //处理亮度为k的星星在（1,1）到（i，j）矩阵内的个数                  }              }          }          for(int i = 0; i < q; i++) {              int t,x1,y1,x2,y2;              scanf("%d%d%d%d%d",&t,&x1,&y1,&x2,&y2);              int ans = 0;              int ans1;              for(int k = 0; k <= c; k++) {                  ans1 = (k+t)%(c+1);        //    计算亮度为k的星星在t秒后的亮度                  int num = sum[x2][y2][k]-sum[x1-1][y2][k]-sum[x2][y1-1][k]+sum[x1-1][y1-1][k];  //计算 （x1,y1）到（x2，y2）矩阵内亮度为k的星星的个数                  ans1 *= num;                  ans += ans1;              }                printf("%d\n",ans);          }      }  }