CodeForces 828B Black Square

题目链接：​​http://codeforces.com/contest/828/problem/B​​​
题意：有一个n*m的二维矩阵，其中W代表白色，B代表黑色，让你把一部分白色变成黑色，使得这个二维矩阵里面有一个黑色正方形，输出改变的个数
解析：确定原图中黑的边界，然后判断是否合法，如果合法就拿黑的面积减去黑的个数，否则输出-1

#include <bits/stdc++.h>
using namespace std;
const int maxn = 105;
char a[maxn][maxn];
int getWidth(int n,int m)
{
    int w1=0,w2=0,flag1 = 0,flag2 = 0;;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(a[i][j]=='B' && !flag1)
            {
                flag1 = 1;
                w1 = i;
            }
            if(a[n-1-i][j]=='B' && !flag2)
            {
                flag2 = 1;
                w2 = n-1-i;
            }
        }
    }
    return w2-w1+1;
}
int getHigh(int n,int m)
{
    int h1=0,h2=0,flag1 = 0,flag2 = 0;;
    for(int i=0;i<m;i++)
    {
        for(int j=0;j<n;j++)
        {
            if(a[j][i]=='B' && !flag1)
            {
                flag1 = 1;
                h1 = i;
            }
            if(a[j][m-1-i]=='B' && !flag2)
            {
                flag2 = 1;
                h2 = m-1-i;
            }
        }
    }
    return h2-h1+1;
}
int main(void)
{
    int n,m;
    scanf("%d %d",&n,&m);
    int cnt = 0;
    for(int i=0;i<n;i++)
    {
        scanf("%s",a[i]);
        for(int j=0;j<m;j++)
        {
            if(a[i][j]=='B')
                cnt++;
        }
    }
    if(cnt==0)
        puts("1");
    else
    {
        int w = getWidth(n,m),h = getHigh(n,m);
        int tt = max(w,h),ans;
        if(tt<=n && tt<=m && tt*tt>=cnt)
            ans = tt*tt-cnt;
        else
            ans = -1;
        printf("%d\n",ans);
    }
    return 0;
}