[Luogu] 四子连棋

https://www.luogu.org/problemnew/show/P2346

广搜

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>

#define maxn 10010
#define inf 1000000007
#define ll long long

using namespace std;
int n, m, ans = inf;

int dx[] = {, , , -, };
int dy[] = {, , , , -};

struct Node {
    int a[][];
    int color;
    int step;
} t;

queue <Node> q;

bool check(Node f) {
    for(int i = ; i <= ; i ++) {
        if(f.a[i][] == f.a[i][] && f.a[i][] == f.a[i][] && f.a[i][] == f.a[i][]) return ;
        if(f.a[][i] == f.a[][i] && f.a[][i] == f.a[][i] && f.a[][i] == f.a[][i]) return ;
    }
    if(f.a[][] == f.a[][] && f.a[][] == f.a[][] && f.a[][] == f.a[][]) return ;
    if(f.a[][] == f.a[][] && f.a[][] == f.a[][] && f.a[][] == f.a[][]) return ;
    return ;
}

void init() { //处理一开始的棋子颜色
    t.step = ;
    t.color = ;
    for(int i = ; i <= ; i ++)
        for(int j = ; j <= ; j ++)
            if(t.a[i][j] == )
                for(int k = ; k <= ; k ++) {
                    int x = i + dx[k];
                    int y = j + dy[k];
                    if(t.a[x][y] == ) continue ;
                    if(x >=  && x <=  && y >=  && y <= ) { //注意边界
                        Node c = t;
                        c.color = t.a[x][y];
                        c.step = ;
                        swap(c.a[i][j], c.a[x][y]);//移动棋子
                        q.push(c);
                    }
                }
}

void bfs() { //广搜模板
    while(!q.empty()) {
        Node b = q.front();
        q.pop();
        if(check(b)) {
            ans = b.step;
            return ;
        }
        for(int i = ; i <= ; i ++)
            for(int j = ; j <= ; j ++) {
                if(b.a[i][j] == ) {
                    for(int k = ; k <= ; k ++) {
                        int x = i + dx[k];
                        int y = j + dy[k];
                        if(x >=  && x <=  && y >=  && y <=  && b.a[x][y] == (b.color ^ )) { //黑白交替移动
                            Node c = b;
                            swap(c.a[i][j], c.a[x][y]);//移动棋子
                            c.color = b.color ^ ;//如果上一步走的是黑，这一步就是白，上一步是白，这一步是黑。
                            c.step = b.step + ;//步数加一
                            q.push(c);//将当前状态入队
                        }
                    }
                }
            }
    }
}

int main() {
    int x, y, z;
    char s[];
    for(int i = ; i <= ; i ++) {
        scanf("%s",s+);
        for(int j = ; j <= ; j ++) { //处理图，便于广搜
            if(s[j] == 'B') t.a[i][j] = ;
            if(s[j] == 'W') t.a[i][j] = ;
            if(s[j] == 'O') t.a[i][j] = ;
        }
    }
    init();
    bfs();
    printf("%d\n", ans);
    return ;
}