55023 - [NOIP2017]棋盘(chess)

解题思路:最短路

空白只能是用魔法弄来的,但它后面的点必须有颜色。 所以我们可以用优先队列记录到每个点的最少花费,原数组不动。

然后直接bfs即可。因为是优先队列,每个点进入队列一次就可以,后面来的花费肯定更多啦。

不太熟悉bfs的同学可以看这里和这里。

代码实现

#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> pi;
typedef pair<pi, pi> pii;
#define f first
#define s second
const int N = 105;
int dir[2][4] = {0, 1, -1, 0, 1, 0, 0, -1};
int G[N][N];
int cost[N][N], a[N][N];
int n, m;
inline bool in(int x, int y) {
    return 1 <= x && x <= n && 1 <= y && 1 <= n;
}
int bfs() {
    priority_queue<pii, vector<pii>, greater<pii> > q;
    q.push(make_pair(make_pair(0, G[1][1]),
                    (make_pair(1, 1))));
    memset(cost, -1, sizeof cost);
    a[1][1] = 1;
    while (q.size()) {
        pii u = q.top(), v;
        q.pop();
        int t1 = u.f.f, t2 = u.f.s;
        int x = u.s.f, y = u.s.s;
        if ((cost[x][y] != -1 && cost[x][y] > t1) || cost[x][y] == -1) {
            cost[x][y] = t1;
        } else {
            continue;
        }
        for (int i = 0; i < 4; ++i) {
            int nxt_x = x + dir[0][i], nxt_y = y + dir[1][i];
            if (G[x][y] == 0 && G[nxt_x][nxt_y] == 0) {
                continue;
            }
            if (!in(nxt_x, nxt_y) || a[nxt_x][nxt_y]) {
                continue;
            }
            if (!G[nxt_x][nxt_y]) {
                v.f.f = t1 + 2;
                v.f.s = t2;
            } else if (G[nxt_x][nxt_y] == t2) {
                v.f.f = t1;
                v.f.s = t2;
            } else {
                v.f.f = t1 + 1;
                v.f.s = G[nxt_x][nxt_y];
            }
            v.s.f = nxt_x;
            v.s.s = nxt_y;
            q.push(v);
            ++a[nxt_x][nxt_y];
        }
    }
    return cost[n][n];
}
int main() {
    cin >> n >> m;
    while (m--) {
        int x, y, c;
        cin >> x >> y >> c;
        G[x][y] = c + 1;
    }
    cout << bfs() << '\n';
    return 0;
}