結果

問題 No.2915 辺更新価値最大化
ユーザー titiatitia
提出日時 2024-11-01 06:54:46
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,904 bytes
コンパイル時間 517 ms
コンパイル使用メモリ 82,428 KB
実行使用メモリ 77,780 KB
最終ジャッジ日時 2024-11-01 06:54:52
合計ジャッジ時間 5,388 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
53,508 KB
testcase_01 AC 39 ms
54,020 KB
testcase_02 AC 47 ms
53,312 KB
testcase_03 AC 39 ms
53,080 KB
testcase_04 AC 39 ms
53,464 KB
testcase_05 AC 41 ms
55,052 KB
testcase_06 AC 37 ms
54,812 KB
testcase_07 AC 92 ms
76,576 KB
testcase_08 AC 107 ms
76,776 KB
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 AC 131 ms
77,432 KB
testcase_22 AC 110 ms
76,876 KB
testcase_23 AC 113 ms
77,508 KB
testcase_24 AC 132 ms
77,592 KB
testcase_25 WA -
testcase_26 AC 129 ms
77,548 KB
testcase_27 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

from heapq import heappop,heappush

N,M,C=map(int,input().split())

EDGE=[list(map(int,input().split())) for i in range(M)]

for i in range(M):
    EDGE[i][0]-=1
    EDGE[i][1]-=1

USE=[0]*N
E=[[] for i in range(N)]

for i in range(M):
    u,v,w=EDGE[i]
    E[u].append((v,i))

Q=[0]
NO=[]
while Q:
    x=Q.pop()
    USE[x]=1
    
    for to,ind in E[x]:
        if USE[to]==1:
            NO.append(ind)
        else:
            USE[to]=1
            Q.append(to)

E=[[] for i in range(N)]
E_INV=[[] for i in range(N)]
NO=set(NO)
for i in range(M):
    if i in NO:
        continue
    u,v,w=EDGE[i]
    E[u].append(v)
    E_INV[v].append(u)


def Top_sort(E):
    Parent=[-1]*N
    USEIND=[0]*N
    TOP=[]

    for ROOT in range(1):
        if Parent[ROOT]!=-1:
            continue
        Parent[ROOT]=ROOT

        NOW=ROOT

        while NOW!=ROOT or USEIND[ROOT]!=len(E[ROOT]):

            if USEIND[NOW]==len(E[NOW]):
                TOP.append(NOW)
                NOW=Parent[NOW]
            elif E[NOW][USEIND[NOW]]==Parent[NOW]:
                USEIND[NOW]+=1
            else:
                NEXT=E[NOW][USEIND[NOW]]
                USEIND[NOW]+=1
                if Parent[NEXT]==-1:
                    Parent[NEXT]=NOW
                    NOW=NEXT
        TOP.append(ROOT)
        
    return TOP[::-1]

TOP=Top_sort(E)


Q=list(map(int,input().split()))

for i in range(C):
    Q[i]-=1

USE=[1]*M

for q in Q:
    USE[q]^=1
    
    DP=[-1<<63]*N
    DP[0]=0

    E=[[] for i in range(N)]

    for i in range(M):
        if USE[i]==0:
            continue
        x,y,dis=EDGE[i]
        E[x].append((y,dis))

    for ind in TOP:
        for to,w in E[ind]:
            if DP[to]<DP[ind]+w:
                DP[to]=DP[ind]+w

    if DP[-1]<-1000000000:
        print("NaN")
    else:
        print(DP[-1])
0