結果

問題 No.2640 traO Stamps
ユーザー rikein12rikein12
提出日時 2024-02-19 23:37:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 769 ms / 2,000 ms
コード長 2,167 bytes
コンパイル時間 185 ms
コンパイル使用メモリ 82,256 KB
実行使用メモリ 105,100 KB
最終ジャッジ日時 2024-09-29 03:14:06
合計ジャッジ時間 20,899 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
53,968 KB
testcase_01 AC 37 ms
53,104 KB
testcase_02 AC 38 ms
53,240 KB
testcase_03 AC 38 ms
53,556 KB
testcase_04 AC 39 ms
54,120 KB
testcase_05 AC 40 ms
53,136 KB
testcase_06 AC 605 ms
95,936 KB
testcase_07 AC 603 ms
95,632 KB
testcase_08 AC 698 ms
101,120 KB
testcase_09 AC 473 ms
91,120 KB
testcase_10 AC 686 ms
100,848 KB
testcase_11 AC 660 ms
93,480 KB
testcase_12 AC 592 ms
103,916 KB
testcase_13 AC 692 ms
93,312 KB
testcase_14 AC 736 ms
101,012 KB
testcase_15 AC 725 ms
93,340 KB
testcase_16 AC 651 ms
100,928 KB
testcase_17 AC 667 ms
100,840 KB
testcase_18 AC 629 ms
93,224 KB
testcase_19 AC 678 ms
104,192 KB
testcase_20 AC 509 ms
101,144 KB
testcase_21 AC 644 ms
93,644 KB
testcase_22 AC 769 ms
103,976 KB
testcase_23 AC 554 ms
91,244 KB
testcase_24 AC 694 ms
91,288 KB
testcase_25 AC 573 ms
104,204 KB
testcase_26 AC 589 ms
105,100 KB
testcase_27 AC 590 ms
105,016 KB
testcase_28 AC 601 ms
104,916 KB
testcase_29 AC 642 ms
104,596 KB
testcase_30 AC 581 ms
103,952 KB
testcase_31 AC 573 ms
104,232 KB
testcase_32 AC 553 ms
98,120 KB
testcase_33 AC 567 ms
100,960 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class BIT:
            
    def __init__(self, L):
        self.N = len(L)
        self.bit = [0]*self.N
        for i,l in enumerate(L):
            self.add(i,l)
        self.N0 = 1
        while self.N0*2 <= self.N:
            self.N0 *= 2

    def add(self, a, w):
        x = a + 1
        for i in range(1000):
            self.bit[x-1] += w
            x += x & -x
            if x > self.N:
                break

    def sum(self, a):
        x = a+1
        ret = 0
        for i in range(1000):
            ret += self.bit[x-1]
            x -= x & -x
            if x <= 0:
                break        
        return ret

    #you can use this function when the BIT has only non-negative values.
    def lower_bound(self, w):
        if w<=0:
            return 0
        x = 0
        k = self.N0
        while k>0:
            if x+k<=self.N:
                if self.bit[x+k-1]<w:
                    w-=self.bit[x+k-1]
                    x+=k
            k//=2
        return x+1
    
N,M,K = list(map(int,input().split()))
S = list(map(int,input().split()))
e_list = [[] for i in range(N)]
for i in range(M):
    u,v,d = list(map(int,input().split()))
    u -= 1
    v -= 1
    e_list[u].append((v,d))
    e_list[v].append((u,d))

INF = float("inf")
D = [[INF]*N for i in range(N)]
for i in range(N):
    D[i][i] = 0
for i in range(N):
    for j,d in e_list[i]:
        D[i][j]=d

for k in range(N):
    for i in range(N):
        for j in range(N):
            if D[i][j]>D[i][k]+D[k][j]:
                D[i][j] = D[i][k]+D[k][j]
S = [s-1 for s in S]
base = [D[S[i]][S[i+1]] for i in range(K)]
bit = BIT(base)

Q = int(input())
for i in range(Q):
    #print(base)
    T,X,Y = list(map(int,input().split()))
    if T==1:
        S[X] = Y - 1
        if X>0:
            bit.add(X-1, D[S[X-1]][S[X]] - base[X-1])
            base[X-1] = D[S[X-1]][S[X]]
        if X<=K-1:
            bit.add(X, D[S[X]][S[X+1]] - base[X])
            base[X] = D[S[X]][S[X+1]]
    else:
        if X>0:
            print(bit.sum(Y-1) - bit.sum(X-1))
        else:
            if Y > 0:
                print(bit.sum(Y-1))
            else:
                print(0)
0