結果

問題 No.2640 traO Stamps
ユーザー Today03
提出日時 2024-02-19 22:38:01
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,651 bytes
コンパイル時間 2,060 ms
コンパイル使用メモリ 200,764 KB
最終ジャッジ日時 2025-02-19 17:22:26
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 5 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#ifdef LOCAL
#include "./debug.cpp"
#else
#define debug(...)
#define print_line
#endif
using namespace std;
using ll = long long;

// #include "kyopro_library/data_structure/fen.cpp"
#include <atcoder/fenwicktree>
using namespace atcoder;

int main() {
    int N, M, K;
    cin >> N >> M >> K;
    vector<int> S(K + 1);
    for (int i = 0; i <= K; i++) cin >> S[i], S[i]--;
    vector<vector<ll>> G(N, vector<ll>(N, INT_MIN));
    for (int i = 0; i < M; i++) {
        int u, v, w;
        cin >> u >> v >> w;
        u--;
        v--;
        G[u][v] = G[v][u] = w;
    }
    for (int i = 0; i < N; i++) G[i][i] = 0;

    for (int k = 0; k < N; k++) {
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (G[i][k] != INT_MIN && G[k][j] != INT_MIN) {
                    G[i][j] = min(G[i][j], G[i][k] + G[k][j]);
                }
            }
        }
    }

    fenwick_tree<ll> fw(K);
    for (int i = 0; i < K; i++) fw.add(i, G[S[i]][S[i + 1]]);

    int Q;
    cin >> Q;
    while (Q--) {
        int t, x, y;
        cin >> t >> x >> y;

        if (t == 1) {
            y--;
            if (x > 0) {
                fw.add(x - 1, -G[S[x - 1]][S[x]]);
                fw.add(x - 1, G[S[x - 1]][y]);
            }
            if (x < K) {
                fw.add(x, -G[S[x]][S[x + 1]]);
                fw.add(x, G[y][S[x + 1]]);
            }
            S[x] = y;
        }

        else {
            if (x == y) {
                cout << 0 << '\n';
            } else {
                cout << fw.sum(x, y) << '\n';
            }
        }

        debug(fw);
    }
}
0