E. Three States - Codeforces Round #327 (Div. 2) 590C States(广搜)

题目大意：有一个M*N的矩阵，在这个矩阵里面有三个王国，编号分别是123，想知道这三个王国连接起来最少需要再修多少路。

分析：首先求出来每个王国到所有能够到达点至少需要修建多少路，然后枚举所有点求出来最少的即可。

代码如下：

---------------------------------------------------------------------------------------------------------

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<queue>
using namespace std;

const int MAXN = 1007;
const int oo = 1e9+7;

int dir[4][2] = { {1,0},{0,1},{-1,0},{0,-1} };
char G[MAXN][MAXN];
struct node
{
    int step[3];
}a[MAXN][MAXN];
struct point
{
    int x, y;
};

void BFS(int M, int N, int k)
{///第k个王国到达所有点的最短距离
    queue<point> Q;
    point q, s;

    for(int i=0; i<M; i++)
    for(int j=0; j<N; j++)
    {
        if(G[i][j] == k+'1')
        {
            q.x = i, q.y = j;
            a[i][j].step[k] = 0;
            Q.push(q);
        }
    }

    while(Q.size())
    {
        q = Q.front();Q.pop();

        for(int i=0; i<4; i++)
        {
            s = q;
            s.x += dir[i][0];
            s.y += dir[i][1];

            if(s.x>=0&&s.x<M && s.y>=0&&s.y<N && G[s.x][s.y] != '#')
            {
                int t = (G[s.x][s.y]=='.' ? 1:0);

                if(a[s.x][s.y].step[k]==-1 || a[q.x][q.y].step[k]+t < a[s.x][s.y].step[k])
                {///因为王国之间没有不用修路，所有可能会回搜
                    a[s.x][s.y].step[k] = a[q.x][q.y].step[k] + t;
                    Q.push(s);
                }
            }
        }
    }
}

int main()
{
    int M, N;

    scanf("%d%d", &M, &N);

    for(int i=0; i<M; i++)
        scanf("%s", G[i]);

    memset(a, -1, sizeof(a));

    for(int i=0; i<3; i++)
        BFS(M, N, i);

    int ans = oo;

    for(int i=0; i<M; i++)
    for(int j=0; j<N; j++)
    {
        if(a[i][j].step[0]!=-1 && a[i][j].step[1]!=-1 && a[i][j].step[2]!=-1)
        {///如果这个点三个王国都能够到达
            int t = (G[i][j]=='.' ? 1:0);
            ans = min(ans, a[i][j].step[0]+a[i][j].step[1]+a[i][j].step[2]-t*2);
        }
    }
    if(ans == oo)
        ans = -1;
    printf("%d\n", ans);

    return 0;
}