結果

問題 No.2640 traO Stamps
コンテスト
ユーザー t98slider
提出日時 2024-03-08 04:07:54
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 70 ms / 2,000 ms
コード長 1,751 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,012 ms
コンパイル使用メモリ 200,680 KB
最終ジャッジ日時 2025-02-20 01:48:43
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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

template <class T> struct fenwick_tree {
    using U = T;

    public:
    fenwick_tree() : _n(0) {}
    fenwick_tree(int n) : _n(n), data(n) {}

    void add(int p, T x) {
        assert(0 <= p && p < _n);
        p++;
        while (p <= _n) {
            data[p - 1] += U(x);
            p += p & -p;
        }
    }

    T sum(int l, int r) {
        assert(0 <= l && l <= r && r <= _n);
        return sum(r) - sum(l);
    }

    private:
    int _n;
    std::vector<U> data;

    U sum(int r) {
        U s = 0;
        while (r > 0) {
            s += data[r - 1];
            r -= r & -r;
        }
        return s;
    }
};

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n, m, k, u, v, w, Q;
    cin >> n >> m >> k;
    vector<int> S(k + 1);
    vector<vector<ll>> A(n, vector<ll>(n, 1ll << 60));
    for(auto &&v : S) cin >> v, v--;
    for(int i = 0; i < m; i++){
        cin >> u >> v >> w;
        u--, v--;
        A[u][v] = A[v][u] = w;
    }
    for(int i = 0; i < n; i++) A[i][i] = 0;
    for(int k = 0; k < n; k++){
        for(int i = 0; i < n; i++){
            for(int j = 0; j < n; j++){
                A[i][j] = min(A[i][j], A[i][k] + A[k][j]);
            }
        }
    }
    fenwick_tree<ll> fw(k);
    for(int i = 0; i < k; i++){
        fw.add(i, A[S[i]][S[i + 1]]);
    }
    cin >> Q;
    while(Q--){
        int T, X, Y;
        cin >> T >> X >> Y;
        if(T == 1){
            Y--;
            if(X >= 1) fw.add(X - 1, A[S[X - 1]][Y] - A[S[X - 1]][S[X]]);
            if(X < k) fw.add(X, A[Y][S[X + 1]] - A[S[X]][S[X + 1]]);
            S[X] = Y;
        }else{
            cout << fw.sum(X, Y) << '\n';
        }
    }
}
0