[IOI 1998] Camelot：最短路

题意：8*8的棋盘上有一个王和0或多个骑士，骑士走日，王上下左右斜，王和骑士相遇后可以骑骑士的马，此后步数不再单独计算。求王、所有骑士汇集在一点的最小总步数。

USACO Training上有强化版，棋盘变为最大26*40。原题可以三次方暴力枚举，加强后怎么做呢？

设汇集点为$P$，骑士的集合为$A$，$a$原位置为$s_a$，$k$代表King，则

$$answer(P) = \sum_{a \in A} dist_{knight}(s_a, P) + \min \{ dist_{knight}(s_b, Q) + dist_{king}(s_k, Q) + dist_{knight}(Q, P)\ (b \in A)\}$$

枚举骑士$b$，用SPFA求出到每一点Q的$\min \{ dist_{knight}(s_b, Q) + dist_{king}(s_k, Q) + dist_{knight}(Q, P)\}$。具体地，给状态加上一维，表示骑士是否已经带上王。没带王可以在此处带王，可以自己继续走；已经带王就带着王继续走。

SPFA的同时求出每个骑士到每一点的最短路，累加即得$\sum_{a \in A} dist_{knight}(s_a, P)$，枚举$P$，得到答案。

原本想枚举骑士接王的位置和骑士，再枚举汇集点。然而，$answer$的表达式$\min$里的部分含汇集点 P <script type="math/tex" id="MathJax-Element-13">P</script>，不知道怎么分离。

既然不知道怎么分离，那就一起算嘛……我也往这方面想过，但是只想到把状态定义为(骑士的位置, 王的位置)。其实，某一时刻骑士和王分别在哪里是没有意义的，可以不出现在状态的表示中。

也可以先求骑士到每点的最短路，然后加上王到该点的最短路，再跑一遍Dijkstra（因为马老师从不写SPFA QAQ）。这样可以不加额外的一维。

有同学猜测骑士不需要绕道接王，这是不对的……网上有一种说法接王的地点仅限于王附近+-1/2，这也是不对的。见http://wiki.codevs.com/wiki/USACO/camelot

#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cassert>
using namespace std;
const int M = 8, N = 8, dr[] = {-2, -1, 1, 2, 2, 1, -1, -2}, dc[] = {1, 2, 2, 1, -1, -2, -2, -1}, inf = 0x3f3f3f3f;
char s[M*N*2+1];
int d[M][N][2], w[M][N], p[M][N], sum[M][N];
bool inq[M][N][2];
struct State {
    int r, c;
    bool f;
};
queue<State> Q;

inline bool in(int r, int c)
{
    return r >= 0 && r < M && c >= 0 && c < N;
}

inline void relax(State v, int dis)
{
    if (d[v.r][v.c][v.f] > dis) {
        d[v.r][v.c][v.f] = dis;
        if (!inq[v.r][v.c][v.f]) {
            inq[v.r][v.c][v.f] = true;
            Q.push(v);
        }
    }
}

// 骑士在(r, c)，到每个点不带王的最短路，带王的最短路
void spfa(int r, int c)
{
    memset(d, 0x3f, sizeof(d));
    relax((State){r, c, false}, 0);
    while (!Q.empty()) {
        State u = Q.front();
        Q.pop();
        inq[u.r][u.c][u.f] = false;
        int dis = d[u.r][u.c][u.f];
        if (!u.f)
            relax((State){u.r, u.c, true}, dis + w[u.r][u.c]);
        for (int k = 0; k < 8; ++k) {
            State v = (State){u.r+dr[k], u.c+dc[k], u.f};
            if (in(v.r, v.c))
                relax(v, dis+1);
        }
    }
}

int main()
{
    memset(p, 0x3f, sizeof(p));
    scanf("%s", s);
    int kr = s[0]-'A', kc = s[1]-'1';
    for (int i = 0; i < M; ++i)
        for (int j = 0; j < N; ++j)
            w[i][j] = max(abs(i-kr), abs(j-kc));
    if (!s[2]) {
        puts("0");
        return 0;
    }
    for (int k = 2; s[k]; k += 2) {
        spfa(s[k]-'A', s[k+1]-'1');
        for (int i = 0; i < M; ++i)
            for (int j = 0; j < N; ++j) {
                p[i][j] = min(p[i][j], d[i][j][1]-d[i][j][0]);
                if (d[i][j][0] == inf)
                    sum[i][j] = inf;
                else if (sum[i][j] != inf)
                    sum[i][j] += d[i][j][0];
            }
    }
    int ans = inf;
    for (int i = 0; i < M; ++i)
        for (int j = 0; j < N; ++j)
            if (sum[i][j] != inf && p[i][j] != inf)
                ans = min(ans, sum[i][j]+p[i][j]);
    printf("%d\n", ans);
    return 0;
}