結果

問題 No.2759 Take Pictures, Elements?
ユーザー Tatsu_mrTatsu_mr
提出日時 2024-05-18 01:01:52
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 667 ms / 2,000 ms
コード長 2,707 bytes
コンパイル時間 2,771 ms
コンパイル使用メモリ 218,780 KB
実行使用メモリ 261,820 KB
最終ジャッジ日時 2024-12-20 15:57:11
合計ジャッジ時間 10,247 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 615 ms
241,924 KB
testcase_01 AC 667 ms
261,820 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 2 ms
6,816 KB
testcase_05 AC 2 ms
6,816 KB
testcase_06 AC 2 ms
6,816 KB
testcase_07 AC 3 ms
6,820 KB
testcase_08 AC 2 ms
6,820 KB
testcase_09 AC 375 ms
226,300 KB
testcase_10 AC 354 ms
226,388 KB
testcase_11 AC 355 ms
226,444 KB
testcase_12 AC 357 ms
226,392 KB
testcase_13 AC 361 ms
226,312 KB
testcase_14 AC 472 ms
227,248 KB
testcase_15 AC 462 ms
227,356 KB
testcase_16 AC 466 ms
227,484 KB
testcase_17 AC 472 ms
227,412 KB
testcase_18 AC 479 ms
227,336 KB
testcase_19 AC 471 ms
227,368 KB
testcase_20 AC 477 ms
227,492 KB
testcase_21 AC 2 ms
6,820 KB
testcase_22 AC 1 ms
6,816 KB
testcase_23 AC 2 ms
6,820 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

struct Edge {
    int to;
    long long cost;
};

struct Dijkstra {
    vector<vector<Edge>> g;
    int n, x;
    vector<long long> dist;
    vector<int> prev;
    
    Dijkstra(vector<vector<Edge>> g_, int x_) : g(g_), n(g.size()), x(x_), dist(n, 1000000000000000000), prev(n) {
        build();
    }
    
    void build() {
        vector<bool> visit(n, false);
        priority_queue<pair<long long, int>, vector<pair<long long, int>>, greater<pair<long long, int>>> que;
        dist[x] = 0LL;
        que.push({dist[x], x});
        while (!que.empty()) {
            int v = que.top().second;
            que.pop();
            if (visit[v]) {
                continue;
            }
            visit[v] = true;
            for (auto e : g[v]) {
                int nv = e.to;
                long long d = e.cost;
                if (dist[nv] > dist[v] + d) {
                    dist[nv] = dist[v] + d;
                    prev[nv] = v;
                    que.push({dist[nv], nv});
                }
            }
        }
    }
    
    long long getdist(int y) {
        return dist[y];
    }
    
    vector<int> getroute(int y) {
        int cur = y;
        vector<int> res = {cur};
        while (cur != x) {
            cur = prev[cur];
            res.push_back(cur);
        }
        reverse(res.begin(), res.end());
        return res;
    }
};

int main() {
    int n, q;
    cin >> n >> q;
    vector<int> a(n), b(q);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    for (int i = 0; i < q; i++) {
        cin >> b[i];
    }
    //(i, j) クエリをi個処理して現在jにいる
    vector<vector<Edge>> g((q + 1) * n);
    for (int i = 0; i <= q; i++) {
        for (int j = 0; j < n - 1; j++) {
            //(i, j) <-> (i, j+1)
            g[n * i + j].push_back({n * i + j + 1, 1LL});
            g[n * i + j + 1].push_back({n * i + j, 1LL});
        }
    }
    for (int i = 1; i <= q; i++) {
        for (int j = 0; j < n; j++) {
            //(i-1, j-1) -> (i, j) (cost 1)
            //(i-1, j)   -> (i, j) (cost 0)
            //(i-1, j+1) -> (i, j) (cost 1)
            if (b[i - 1] == a[j]) {
                if (j >= 1) {
                    g[n * (i - 1) + j - 1].push_back({n * i + j, 1LL});
                }
                g[n * (i - 1) + j].push_back({n * i + j, 0LL});
                if (j < n - 1) {
                    g[n * (i - 1) + j + 1].push_back({n * i + j, 1LL});
                }
            }
        }
    }
    Dijkstra D(g, 0);
    long long ans = 1000000000000000000;
    for (int j = 0; j < n; j++) {
        ans = min(ans, D.getdist(n * q + j));
    }
    cout << ans << endl;
}
0