电路维修 (广搜变形-双端队列bfs)

# 2632. 「BalticOI 2011 Day1」打开灯泡 Switch the Lamp On

【题目描述】

有一种正方形的电路元件,在它的两组相对顶点中,有一组会用导线连接起来,另一组则不会。
有 $M\times N$ 个这样的元件,你想将其排列成 $N$ 行 $M$ 列放在电路板上。电路板的左上角连接电源,右下角连接灯泡。

试求:至少要旋转多少个正方形元件才能让电源与灯泡连通,若无解则输出 NO SOLUTION。

【算法】

双端队列bfs,最短路的变形。注意更新判重时可能存在对最短路后更新的情况。同时记得用常数数组减少if语句,简化代码。

【代码】

#include <bits/stdc++.h>
#define P pair<int,int>
#define ff first
#define ss second
using namespace std;
int n,m;
int d[510][510];
char s[510][510];
const int dx[4]={-1,1,1,-1},dy[4]={1,1,-1,-1},val[4]={'/','\\','/','\\'};
const int bias_x[4]={-1,0,0,-1},bias_y[4]={0,0,-1,-1};
deque<P> q;
int main() {
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) {
        scanf("%s",s[i]+1);
    }
    for(int i=1;i<=n+1;i++)
        for(int j=1;j<=m+1;j++)
            d[i][j]=-1;
    d[1][1]=0; q.push_back(make_pair(1,1));
    while(q.size()) {
        P now=q.front(); q.pop_front();
        if(now.ff==n+1&&now.ss==m+1) break;
        for(int i=0;i<4;i++) {
            int nx=now.ff+dx[i],ny=now.ss+dy[i];
            if(nx>=1&&nx<=n+1&&ny>=1&&ny<=m+1&&(d[nx][ny]==-1||d[nx][ny]>=d[now.ff][now.ss]+1)) {
                int xx=now.ff+bias_x[i],yy=now.ss+bias_y[i];
                if(s[xx][yy]==val[i]) {
                    d[nx][ny]=d[now.ff][now.ss];
                    q.push_front(make_pair(nx,ny));
                }else {
                    d[nx][ny]=d[now.ff][now.ss]+1;
                    q.push_back(make_pair(nx,ny));
                }
            }
        }
    }
    if(d[n+1][m+1]==-1) puts("NO SOLUTION");
    else printf("%d\n",d[n+1][m+1]);
    return 0;
}

posted @ 2018-09-01 23:49  飞飞翔滴少年  阅读(802)  评论(0编辑  收藏  举报