#595737 (C++17(clang)) No.1323 うしらずSwap

提出ソース

結果

問題	No.1323 うしらずSwap
ユーザー	t33f
提出日時	2020-12-20 12:38:28
言語	C++17(clang) (17.0.6 + boost 1.83.0)
結果	WA
実行時間	-
コード長	6,996 bytes
コンパイル時間	1,409 ms
コンパイル使用メモリ	133,088 KB
実行使用メモリ	35,580 KB
最終ジャッジ日時	2024-09-21 11:41:23
合計ジャッジ時間	7,500 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,940 KB
testcase_02	AC	1 ms 6,940 KB
testcase_03	AC	2 ms 6,944 KB
testcase_04	AC	2 ms 6,940 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	2 ms 6,944 KB
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	AC	1 ms 6,944 KB
testcase_11	WA	-
testcase_12	AC	2 ms 6,940 KB
testcase_13	AC	2 ms 6,944 KB
testcase_14	WA	-
testcase_15	WA	-
testcase_16	AC	65 ms 35,388 KB
testcase_17	AC	101 ms 35,320 KB
testcase_18	AC	64 ms 35,472 KB
testcase_19	AC	94 ms 35,476 KB
testcase_20	AC	63 ms 35,304 KB
testcase_21	AC	82 ms 35,436 KB
testcase_22	AC	133 ms 35,380 KB
testcase_23	AC	81 ms 35,280 KB
testcase_24	AC	134 ms 35,404 KB
testcase_25	AC	82 ms 35,320 KB
testcase_26	AC	77 ms 35,536 KB
testcase_27	AC	80 ms 35,412 KB
testcase_28	AC	80 ms 35,324 KB
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	AC	128 ms 35,200 KB
testcase_34	AC	125 ms 35,220 KB
testcase_35	AC	118 ms 35,368 KB
testcase_36	AC	116 ms 35,484 KB
testcase_37	AC	113 ms 35,320 KB
testcase_38	AC	110 ms 35,120 KB
testcase_39	AC	116 ms 35,216 KB
testcase_40	AC	115 ms 35,360 KB
testcase_41	AC	117 ms 35,428 KB
testcase_42	AC	117 ms 35,320 KB
testcase_43	AC	58 ms 35,368 KB
testcase_44	AC	77 ms 35,316 KB
testcase_45	AC	69 ms 35,432 KB
testcase_46	AC	77 ms 35,260 KB
testcase_47	AC	62 ms 35,452 KB
testcase_48	WA	-
testcase_49	WA	-
testcase_50	WA	-
testcase_51	WA	-
testcase_52	WA	-
testcase_53	WA	-
testcase_54	WA	-
testcase_55	WA	-
testcase_56	WA	-
testcase_57	WA	-
testcase_58	WA	-
testcase_59	AC	3 ms 6,944 KB
testcase_60	AC	2 ms 6,940 KB
testcase_61	WA	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <cassert>
#include <cstring>
#include <tuple>
#include <algorithm>
#include <queue>
#include <vector>
#include <iostream>
using namespace std;
template<typename Callback>
inline void for_nbd4(int x, int y, int lx, int ux, int ly, int uy, Callback f) {
    constexpr pair<int, int> dxdy[] = {{1, 0}, {0, 1}, {-1, 0}, {0, -1}};
    for (auto [dx, dy] : dxdy) {
        const int u = x + dx, v = y + dy;
        if (lx <= u && u < ux && ly <= v && v < uy) f(u, v);
    }
}
template<typename Callback>
inline void for_nbd4(int x, int y, int X, int Y, Callback f) {
    for_nbd4(x, y, 0, X, 0, Y, f);
}

template<typename T, typename Callback>
inline T sum_nbd4(int x, int y, int lx, int ux, int ly, int uy, Callback f) {
    constexpr pair<int, int> dxdy[] = {{1, 0}, {0, 1}, {-1, 0}, {0, -1}};
    T result{};
    for (auto [dx, dy] : dxdy) {
        const int u = x + dx, v = y + dy;
        if (lx <= u && u < ux && ly <= v && v < uy) result += f(u, v);
    }
    return result;
}
template<typename T, typename Callback>
inline T sum_nbd4(int x, int y, int X, int Y, Callback f) {
    return sum_nbd4<T, Callback>(x, y, 0, X, 0, Y, f);
}

constexpr int INF = 1<<30;

int solve(vector<string> S, int sx, int sy, int gx, int gy) {
    const int h = S.size(), w = S.front().size();
    int dist[h][w], count[h][w];
    pair<int, int> from[h][w];
    queue<tuple<int, int> > q;
    q.emplace(sx, sy);
    for (int i = 0; i < h; i++)
        for (int j = 0; j < w; j++)
            if (i == sx && j == sy) {
                dist[i][j] = 0;
                count[i][j] = 1;
                from[i][j] = {-1, -1};
            } else {
                dist[i][j] = INF;
                count[i][j] = 0;
            }
    while (!q.empty()) {
        auto [x, y] = q.front();
        q.pop();
        for_nbd4(x, y, h, w, [x=x, y=y, &S, &dist, &q, &count, &from](int u, int v) {
                if (S[u][v] == '#') return;
                if (dist[u][v] > dist[x][y] + 1) {
                    dist[u][v] = dist[x][y] + 1;
                    count[u][v] = count[x][y];
                    from[u][v] = {x, y};
                    q.emplace(u, v);
                } else if (dist[u][v] == dist[x][y] + 1) {
                    assert(count[u][v] > 0);
                    assert(count[x][y] > 0);
                    count[u][v] = 2;
                }
            });
    }
    // for (int i = 0; i < h; i++)
    //     for (int j = 0; j < w; j++)
    //         cerr << i << ' ' << j << ' ' << dist[i][j] << ' ' << count[i][j] << endl;
    auto is_empty = [&S](int u, int v) -> int {return int(S[u][v] == '.');};

    if (count[gx][gy] == 0) return -1;
    else if (count[gx][gy] > 1) return 2 * dist[gx][gy];
    else {
        for (int x = gx, y = gy;;) {
            tie(x, y) = from[x][y];
            if (x == sx && y == sy) break;
            if (sum_nbd4<int>(x, y, h, w, is_empty) > 2) {
                return 2 * dist[gx][gy] + 2;
            }
        }

        int ans = INF;
        if (sum_nbd4<int>(gx, gy, h, w, is_empty) == 4
            || sum_nbd4<int>(sx, sy, h, w, is_empty) == 4) {
            ans = 2 * dist[gx][gy] + 4;
        }

        for (int x = gx, y = gy;;) {
            tie(x, y) = from[x][y];
            if (x == sx && y == sy) break;
            S[x][y] = '#';
        }

        const int d0 = dist[gx][gy];

        for (int i = 0; i < h; i++) fill(dist[i], dist[i] + w, INF);
        for_nbd4(sx, sy, h, w, [gx, gy, &q, &dist, &S](int u, int v) {
                if (S[u][v] == '.' && (u != gx || v != gy)) {
                    q.emplace(u, v);
                    dist[u][v] = 1;
                }
            });
        while (!q.empty()) {
            auto [x, y] = q.front();
            q.pop();
            for_nbd4(x, y, h, w, [x=x, y=y, sx, sy, gx, gy, &S, &dist, &q](int u, int v) {
                    if (S[u][v] == '#') return;
                    if (u == sx && v == sy) return;
                    if (dist[u][v] > dist[x][y] + 1) {
                        dist[u][v] = dist[x][y] + 1;
                        q.emplace(u, v);
                    }
                });
        }

        ans = min(ans, d0 + dist[gx][gy]);
        return (ans == INF ? -1 : ans);
    }
}

int dp(const vector<string>& S, int sx, int sy, int gx, int gy) {
    const int h = S.size(), w = S.front().size();
    int dp[h][w][h][w];
    for (int a = 0; a < h; a++)
        for (int b = 0; b < w; b++)
            for (int c = 0; c < h; c++)
                for (int d = 0; d < w; d++)
                    dp[a][b][c][d] = INF;
    queue<tuple<int, int, int, int> > q;
    dp[sx][sy][gx][gy] = 0;
    
    q.emplace(sx, sy, gx, gy);
    while (!q.empty()) {
        auto [a, b, c, d] = q.front();
        q.pop();
        const int dist = dp[a][b][c][d];
        for_nbd4(a, b, h, w, [c=c, d=d, dist, &dp, &S, &q](int u, int v) {
                if (S[u][v] == '#') return;
                if (c == u && d == v) return;
                if (dp[u][v][c][d] > dist + 1) {
                    dp[u][v][c][d] = dist + 1;
                    q.emplace(u, v, c, d);
                }
            });
        for_nbd4(c, d, h, w, [a=a, b=b, dist, &dp, &S, &q](int u, int v) {
                if (S[u][v] == '#') return;
                if (a == u && b == v) return;
                if (dp[a][b][u][v] > dist + 1) {
                    dp[a][b][u][v] = dist + 1;
                    q.emplace(a, b, u, v);
                }
            });
    }
    return dp[gx][gy][sx][sy] == INF ? -1 : dp[gx][gy][sx][sy];
}

int main() {
    int h, w, sx, sy, gx, gy; cin >> h >> w >> sx >> sy >> gx >> gy;
    sx--; sy--; gx--; gy--;
    vector<string> S(h);
    for (int i = 0; i < h; i++) cin >> S[i];
    cout << solve(S, sx, sy, gx, gy) << endl;

    // int h = 5, w = 8;
    // vector S(6, string(6, '#'));
    // const int k = (h - 2) * (w - 2);
    // for (int s = 0; s < 1<<k; s++) {
    //     if ((s & 1023) == 0) cerr << "s = " << s << " of " << (1<<k) << endl;
    //     for (int idx = 0, i = 1; i < h-1; i++)
    //         for (int j = 1; j < w-1; j++) {
    //             S[i][j] = (((s >> idx) & 1) == 1 ? '#' : '.');
    //         }
    //     for (int a = 1; a < h-1; a++)
    //         for (int b = 1; b < w-1; b++) {
    //             if (S[a][b] == '#') continue;
    //             for (int c = 1; c < h-1; c++)
    //                 for (int d = 1; d < w-1; d++) {
    //                     if (S[c][d] == '#') continue;
    //                     if (a == c && b == d) continue;
    //                     int ans = solve(S, a, b, c, d),
    //                         ans2 = dp(S, a, b, c, d);
    //                     // cout << ans << ' ' << ans2 << endl;
    //                     if(ans != ans2) {
    //                         cerr << a << ' ' << b << ' ' << c << ' ' << d << endl;
    //                         for (string &line : S) cerr << line << endl;
    //                     }
    //                 }
    //         }
    // }
}

yukicoder

結果

テストケース

ソースコード