牛客寒假算法基础集训营6 E 海啸

题目链接点这里

这个题输入类型是第一次见，并没有把n和m的具体范围给你，但是给了n*m的范围，武断的设为1e6*1e6的二维数组铁铁WA，就将二维数组转换为一维数组

题目类型属于二维数组前缀和，有um[i][j]=sum[i−1][j]+sum[i][j−1]−sum[i−1][j−1]+a[i][j]

求矩阵(a,b)到(x,y)的矩阵的和可以用sum[x][y]−sum[x][b−1]−sum[a−1][y]+sum[a−1][b−1]

在前期处理的时候，将水位超过d的设为1，否则设为0

代码如下：

#include <bits/stdc++.h>
using namespace std;
const int inf=<<;
typedef long long ll;
const double pi=acos(-);
const int mod=1e9+;
const int maxn=2e6;
int h[maxn];
int sum[maxn];
int main(){
    int n,m,d;scanf("%d%d%d",&n,&m,&d);
    for(int i=;i<=n;i++){
        for(int j=;j<=m;j++){
            int t;scanf("%d",&t);
            t=t>=d?:;
            h[i*(m+)+j]=t;
            sum[i*(m+)+j]=sum[(i-)*(m+)+j]+sum[i*(m+)+j-]-sum[(i-)*(m+)+j-]+h[i*(m+)+j];
        }
    }
    int q;scanf("%d",&q);
    while(q--){
        int a,b,x,y;scanf("%d%d%d%d",&a,&b,&x,&y);
        printf("%d\n",sum[x*(m+)+y]-sum[(a-)*(m+)+y]-sum[x*(m+)+b-]+sum[(a-)*(m+)+b-]);
    }
    return ;
}