牛客寒假算法基础集训营6 E. 海啸(二维数组+容斥)

时间:2022-06-18
本文章向大家介绍牛客寒假算法基础集训营6 E. 海啸(二维数组+容斥),主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

题目链接:https://ac.nowcoder.com/acm/contest/332/E

       二维数组维护前缀和(pre[i][j] = pre[i][j] + pre[i-1][j] + pre[i][j-1] - pre[i-1][j-1]),然后根据O(1)的复杂度就能算出范围内所需要的数了,给的数据范围可能没法开数组,但是可以用vector去存,还有就是用一种不太提倡的方法去开,像我的代码一样(官方题解说开一个指针数组,然后new一个...不会)。其他的主要就是一个容斥原理。


AC代码:

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int n,m,d,t;
 
int main()
{
  scanf("%d%d%d",&n,&m,&d);
  int pre[n + 5][m + 5];
  int ans = 0;
  memset(pre,0,sizeof(pre));
  for(int i=1;i<=n;i++){
    for(int j=1;j<=m;j++){
      int x;
      scanf("%d", &x);
      if(x >= d)pre[i][j] = 1 + pre[i-1][j] + pre[i][j-1] - pre[i-1][j-1];
      else pre[i][j] = pre[i-1][j] + pre[i][j-1] - pre[i-1][j-1];
    }
  }
  scanf("%d",&t);
  while(t--){
    int x1,y1,x2,y2;
    scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
    printf("%dn", pre[x2][y2] - pre[x2][y1-1] - pre[x1-1][y2] + pre[x1-1][y1-1]);
  }
  return 0;
}