#952661 (C++17) No.2640 traO Stamps

• 問題: No.2640 traO Stamps
• ユーザー: Nzt3
• 提出日時: 2024-02-19 21:49:30
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,307 bytes
• コンパイル時間: 2,460 ms
• コンパイル使用メモリ: 208,984 KB
• 実行使用メモリ: 6,824 KB
• 最終ジャッジ日時: 2024-09-29 01:47:07
• 合計ジャッジ時間: 6,626 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,816 KB
testcase_01	AC	2 ms; 6,820 KB
testcase_02	AC	2 ms; 6,820 KB
testcase_03	AC	2 ms; 6,820 KB
testcase_04	AC	2 ms; 6,820 KB
testcase_05	AC	2 ms; 6,820 KB
testcase_06	AC	60 ms; 6,820 KB
testcase_07	AC	58 ms; 6,820 KB
testcase_08	AC	67 ms; 6,816 KB
testcase_09	AC	52 ms; 6,816 KB
testcase_10	AC	64 ms; 6,816 KB
testcase_11	AC	52 ms; 6,820 KB
testcase_12	AC	63 ms; 6,816 KB
testcase_13	WA	-
testcase_14	AC	67 ms; 6,816 KB
testcase_15	WA	-
testcase_16	AC	55 ms; 6,820 KB
testcase_17	AC	57 ms; 6,816 KB
testcase_18	AC	53 ms; 6,824 KB
testcase_19	AC	59 ms; 6,816 KB
testcase_20	AC	66 ms; 6,816 KB
testcase_21	WA	-
testcase_22	AC	68 ms; 6,820 KB
testcase_23	AC	52 ms; 6,820 KB
testcase_24	AC	57 ms; 6,820 KB
testcase_25	AC	64 ms; 6,816 KB
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	AC	74 ms; 6,816 KB
testcase_31	AC	73 ms; 6,820 KB
testcase_32	AC	69 ms; 6,820 KB
testcase_33	AC	70 ms; 6,820 KB

ソースコード

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
namespace Lib {
struct BIT {
  vector<ll> a;
  int sz;
  BIT(int n) : sz(n), a(n + 1) {}
  void add(int p, int v) {
    for (p += 1; p <= sz; p += p & -p) a[p] += v;
  }
  ll sum(int l, int r) {
    ll ret = 0;
    for (; r > 0; r -= r & -r) ret += a[r];
    for (; l > 0; l -= l & -l) ret -= a[l];
    return ret;
  }
};
}  // namespace Lib

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int N,M,K;
  cin>>N>>M>>K;
  vector S(K+1,0);
  for(int &i:S)cin>>i,--i;
  vector dist(N,vector(N,(ll)1e18));
  for(int i=0;i<M;i++){
    int A,B,C;
    cin>>A>>B>>C;
    --A,--B;
    dist[A][B]=C;
    dist[B][A]=C;
  }
  for(int i=0;i<N;i++)dist[i][i]=0;
  for(int i=0;i<N;i++){
    for(int j=0;j<N;j++){
      for(int k=0;k<N;k++){
        dist[j][k]=min(dist[j][k],dist[j][i]+dist[i][k]);
      }
    }
  }

  Lib::BIT seg(K);
  for(int i=0;i<K;i++){
    seg.add(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;
      if(X>0)seg.add(X-1,-dist[S[X-1]][S[X]]);
      if(X<K)seg.add(X,-dist[S[X]][S[X+1]]);
      S[X]=Y;
      if(X>0)seg.add(X-1,dist[S[X-1]][S[X]]);
      if(X<K)seg.add(X,dist[S[X]][S[X+1]]);
    }else{
      cout<<seg.sum(X,Y)<<'\n';
    }
  }
}