【AT ABC337】D-Cheating Gomoku Narabe

 暴力求解显然会TLE

既然要求k长度段内“ . ”的最小个数，我们不妨用前缀和预处理每行每列出现的" . "以及“ x ”的个数。如果这一段有x那么就跳到下一段。

/*I lv ya*/
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <vector>
#include <stack>
#include <sstream>
#include <map>
#include <set>
#include <queue>
#include <stdlib.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N = 2e5+5;
int n,m,k;
char ch;
void solve()
{
    cin>>n>>m>>k;
    int mp[n+5][m+5];
    memset(mp,0,sizeof(mp));
    int s11[n+5][m+5],s12[n+5][m+5];
    int s21[n+5][m+5],s22[n+5][m+5];///前缀和优化
    memset(s11,0,sizeof(s11));
    memset(s12,0,sizeof(s12));
    memset(s21,0,sizeof(s21));
    memset(s22,0,sizeof(s22));
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            cin>>ch;
            if(ch=='o')
            {
                mp[i][j]=1;
            }
            else if(ch=='x')
            {
                mp[i][j]=2;
            }
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            s11[i][j]=s11[i][j-1]+(mp[i][j]==0);
            s12[i][j]=s12[i][j-1]+(mp[i][j]==2);
        }
    }
    for(int i=1;i<=m;i++)
    {
        for(int j=1;j<=n;j++)
        {
            s21[j][i]=s21[j-1][i]+(mp[j][i]==0);
            s22[j][i]=s22[j-1][i]+(mp[j][i]==2);
        }
    }
    int ans=1e9;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m-k+1;j++)
        {
            if(s12[i][j+k-1]-s12[i][j-1]==0)
            {
                ans=min(ans,s11[i][j+k-1]-s11[i][j-1]);
            }
        }
    }
    for(int i=1;i<=m;i++)
    {
        for(int j=1;j<=n-k+1;j++)
        {
            if(s22[j+k-1][i]-s22[j-1][i]==0)
            {
                ans=min(ans,s21[j+k-1][i]-s21[j-1][i]);
            }
        }
    }
    if(ans<1e9)
        cout<<ans<<endl;
    else
        cout<<-1<<endl;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL),cout.tie(NULL);
    int _;
    _=1;
    while(_--)
    {
        solve();
    }
    return 0;
}