cf D. Broken Monitor

http://codeforces.com/contest/370/problem/D

题意：输入一张图，上面只有两个字符'w'和‘.’ ,如果可以用一个正方形把所有的‘w’围起来，所有的‘w’都在正方形的边上。如果有多种输出最小的一个。

先预处理出[1,1]到[i,j]里面有多少个'w'存在dp[i][j]中。找到正方形的大小，然后枚举找左上角的点。就可以找到符合题意的正方形。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 2001
using namespace std;

char g[maxn][maxn];
int dp[maxn][maxn];
int n,m;

int get_num(int x1,int y1,int x2,int y2)
{
    if(x1>x2||y1>y2) return 0;
    return dp[x2][y2]-dp[x2][y1-1]-dp[x1-1][y2]+dp[x1-1][y1-1];
}


int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        int max1=0,max2=0,min1=n,min2=m;
        int cnt=0;
        for(int i=1; i<=n; i++)
        {
            scanf("%s",g[i]+1);
            for(int j=1; j<=m; j++)
            {
                if(g[i][j]=='w')
                {
                    cnt++;
                    min1=min(min1,i);
                    min2=min(min2,j);
                    max1=max(max1,i);
                    max2=max(max2,j);
                }
            }
        }
        bool flag=false;
        for(int i=min1+1; i<max1; i++)
        {
            for(int j=min2+1; j<max2; j++)
            {
                if(g[i][j]=='w')
                {
                    flag=true;
                    break;
                }
            }
            if(flag) break;
        }
        if(flag)
        {
            printf("-1
");
        }
        else
        {
            int x1,y1;
            bool flag1=false;
            int dx=max1-min1+1;
            int dy=max2-min2+1;
            int size=max(dx,dy);
            for(int i=1; i<=n; i++)
            {
                for(int j=1; j<=m; j++)
                {
                    dp[i][j]=dp[i][j-1]+dp[i-1][j]-dp[i-1][j-1];
                    if(g[i][j]=='w')
                    {
                        dp[i][j]+=1;
                    }
                }
            }
            for(int i=1; i<=n; i++)
            {
                if(i+size-1>n)break;
                for(int j=1; j<=m; j++)
                {
                    if(j+size-1>m) break;
                    if(cnt==get_num(i,j,i+size-1,j+size-1)-get_num(i+1,j+1,i+size-2,j+size-2))
                    {
                        x1=i;
                        y1=j;
                        flag1=true;
                        break;
                    }
                }
                if(flag1) break;
            }
            if(flag1)
            {
                for(int i=x1; i<x1+size; i++)
                {
                    for(int j=y1; j<y1+size; j++)
                    {
                        if((g[i][j]=='.'&&i==x1)||(g[i][j]=='.'&&j==y1)||(g[i][j]=='.'&&i==x1+size-1)||(g[i][j]=='.'&&j==y1+size-1)) g[i][j]='+';
                    }
                }
                for(int i=1; i<=n; i++)
                {
                    for(int j=1; j<=m; j++)
                    {
                        printf("%c",g[i][j]);
                    }
                    printf("
");
                }
            }
            else
            {
                printf("-1
");
            }
        }
    }
    return 0;
}