【Codeforces Round #460 (Div. 2) C】 Seat Arrangements

【链接】 我是链接,点我呀:)
【题意】

在这里输入题意

【题解】

用pre[i][j]表示第i行前j列的和。 然后枚举连续座位的最左上点。 (有两种可能向右或向下k个。 则还需要处理出pre2[i][j]表示第j列前i行的和。 (都可以O(N^2)求出

然后pre[i][j+k-1]-pre[i][j-1]就是第i行第j列往右k个格子的空(或被占据)的格子数目了。
看看是不是符合要求就好。
往下扩展同理。

k==1的时候只要取向右一种就好。所以直接除个2

【代码】

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

const int N = 2e3;

int n,m,k;
int a[N+10][N+10],pre[N+10][N+10];
char s[N+10][N+10];

int main(){
	#ifdef LOCAL_DEFINE
	    freopen("rush_in.txt", "r", stdin);
	#endif
	ios::sync_with_stdio(0),cin.tie(0);
    cin >> n >> m >> k;
    for (int i = 1;i <= n;i++){
        cin >>(s[i]+1);
    }
    for (int i = 1;i <= n;i++)
        for (int j = 1;j <= m;j++)
            if (s[i][j]=='*')
                a[i][j] = 1;
            else
                a[i][j] = 0;
    for (int i = 1;i <= n;i++){
        pre[i][0] = 0;
        for (int j = 1;j <=m;j++)
            pre[i][j] = pre[i][j-1]+a[i][j];
    }

    long long ans = 0;
    for (int i = 1;i <=n;i++)
        for (int j = 1;j <= m;j++){
            int l = j+k-1;
            if (l <= m){
                int temp = pre[i][l]-pre[i][j-1];
                if (temp==0){
                    ans++;
                }
            }
        }

    memset(pre,0,sizeof pre);
    for (int j = 1;j <= m;j++){
        pre[0][j] = 0;
        for (int i = 1;i <= n;i++){
            pre[i][j] = pre[i-1][j] + a[i][j];
        }
    }

    for (int i = 1;i <= n;i++)
        for (int j = 1;j <= m;j++){
            int l = i + k-1;
            if (l <= n){
                int temp = pre[l][j]-pre[i-1][j];
                if (temp==0){
                    ans++;
                }
            }
        }
    if (k==1) ans/=2;
    cout<<ans<<endl;
	return 0;
}