結果

問題 No.2640 traO Stamps
コンテスト
ユーザー noya2
提出日時 2024-02-13 23:15:24
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 76 ms / 2,000 ms
コード長 1,469 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 5,271 ms
コンパイル使用メモリ 313,620 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-09-28 18:40:29
合計ジャッジ時間 9,942 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using ll = long long;
const ll linf = 1e18;

template<typename T>
bool chmin(T &a, const T &b){
    if (a <= b) return false;
    a = b;
    return true;
}

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, m, k; cin >> n >> m >> k;
    vector<vector<ll>> dist(n,vector<ll>(n,linf));
    for (int i = 0; i < n; i++){
        dist[i][i] = 0;
    }
    vector<int> s(k+1);
    for (int i = 0; i <= k; i++){
        int v; cin >> v; v--;
        s[i] = v;
    }
    for (int i = 0; i < m; i++){
        int u, v; cin >> u >> v; u--, v--;
        ll w; cin >> w;
        chmin(dist[u][v],w);
        chmin(dist[v][u],w);
    }
    for (int l = 0; l < n; l++) for (int i = 0; i < n; i++) for (int j = 0; j < n; j++){
        chmin(dist[i][j],dist[i][l]+dist[l][j]);
    }
    atcoder::fenwick_tree<ll> fen(k);
    auto set = [&](int i, ll x){
        ll d = x - fen.sum(i,i+1);
        fen.add(i,d);
    };
    for (int i = 0; i < k; i++){
        set(i,dist[s[i]][s[i+1]]);
    }
    int q; cin >> q;
    while (q--){
        int t, x, y; cin >> t >> x >> y;
        if (t == 1){
            y--;
            s[x] = y;
            if (x > 0){
                set(x-1,dist[s[x-1]][s[x]]);
            }
            if (x < k){
                set(x,dist[s[x]][s[x+1]]);
            }
        }
        if (t == 2){
            cout << fen.sum(x,y) << '\n';
        }
    }
}
0