wnnafly挑战赛3 D-Butterfly2

题目描述

给定一个 n*m 的矩阵，矩阵元素由X和O构成，请求出其中最大的蝴蝶形状。
蝴蝶形状的定义如下：
存在一个中心点（必须为X），并且其往左上（必须为X）、左下（必须为X）、右上（必须为O）、右下（必须为O）四个方向扩展相同的长度，且左上顶点与左下顶点之间全由X填充，右上顶点与右下顶点之间全由O填充。

我们不在意在蝴蝶形状内部是X还是O。 

例如：
    XAAAO
    XXAOO
    XAXAO
    XXAOO
    XAAAO
是一个蝴蝶形状（其中A表示X或O）。
     X
也是。
而
    XAAO
    XXOO
    XXOO
    XAAO
不是（不存在中心点）。

输入描述:

第一行两个整数n, m表示矩阵的大小 (1 <= n, m <= 500)；
接下来 n 行，每行一个长度为 m 的字符串表示矩阵，矩阵元素保证由X和O构成。

输出描述:

一行一个整数表示最大的蝴蝶形状的对角线的长度。

示例1

输入

5 5
XOOOO
XXOOO
XOXOO
XXOOO
XOOOO

输出

5

还以为跟Butterfly1一样难。。结果只有500个点。。枚举各个X点，向4各方向扩散判断就好了

500个点。。暴力 500 * 500 * 500。。。 复杂度罩得住。

#include <iostream>
#include <algorithm>
using namespace std;
int n, m;
char map[505][505];
int dir[4][2] = {{-1, -1}, {-1, 1}, {1, -1}, {1, 1}};
inline bool check(int x, int y, char ch) {
    if (map[x][y] == ch)
        return false;
    return true;
}

int check_out(int x, int y) {
    if (map[x][y] != 'X')
        return 0;
    int res = 1;
    //扩散范围
    int s = min(x - 1, y - 1);
    s = min(s, min(n - x, m - y));
    int tx[4], ty[4];
    for (int i = 0; i < 4; ++i) {
        tx[i] = x;
        ty[i] = y;
    }
    for (int i = 1; i <= s; ++i) {
        for (int j = 0; j < 4; ++j) {
            tx[j] += dir[j][0];
            ty[j] += dir[j][1];
        }
        int flag = check(tx[0], ty[0], 'O') & check(tx[1], ty[1], 'X') & check(tx[2], ty[2], 'O') & check(tx[3], ty[3], 'X');
            //cout << x << "  " << y << "   "  << check(tx[1], tx[1], 'X') << endl;
        if (!flag)
            return res;
        else {
            int f = 0;
            for (int j = tx[0]; j <= tx[2]; ++j) {
                if (f)
                    break;
                if (map[j][ty[0]] != 'X')
                    f = 1;
                if (map[j][ty[1]] != 'O')
                    f = 1;
            }
            if (!f)
                res = max(res, tx[2] - tx[0] + 1);
        }
            
    }
    return res;
}
int main() {
        //freopen("in.txt", "r", stdin);
    cin >> n >> m;
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= m; ++j) {
            cin >> map[i][j];
        }
    }
    int ans = 0;
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= m; ++j) {
            ans = max(ans, check_out(i, j));
        }
    }
    cout << ans << endl;
}