結果

問題 No.872 All Tree Path
ユーザー 6soukiti296soukiti29
提出日時 2019-08-30 23:06:55
言語 Nim
(2.0.2)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,261 bytes
コンパイル時間 1,479 ms
コンパイル使用メモリ 65,664 KB
最終ジャッジ日時 2024-07-02 16:00:52
合計ジャッジ時間 3,199 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
/home/judge/data/code/Main.nim(18, 35) Error: type mismatch: got 'seq[int]' for 'map(split(readLine(stdin), {' ', '\t', '\v', '\r', '\n', '\f'}, -1), parseInt)' but expected 'tuple'

ソースコード

diff # 

import strutils,sequtils,deques

type
    edge = tuple[R,L,w : int]
var
    N = stdin.readline.parseInt
    que = initDeque[int](1024)
    ln,rn : int
    used : array[200010, bool]
    cnt : array[200010, int]
    cnte : array[200010, int]
    edges = newSeq[edge](N)
    v,u,w : int
    nodes = newSeqWith(N + 1, newSeq[int](0))
    ans : int64

for n in 0..(N - 2):
    (v,u,w) = stdin.readline.split.map(parseInt)
    v -= 1
    u -= 1
    edges[n] = (v, u, w)
    cnte[v] += 1
    cnte[u] += 1
    nodes[v].add(n)
    nodes[u].add(n)



for n in 0..<N:
    if cnte[n] == 1:
        que.addLast(nodes[n][0])
    cnt[n] = 1

while que.len > 0:
    var e1 = que.popFirst
    if used[e1]:
        continue
    var e = edges[e1]
    used[e1] = true
    if cnte[e.R] == 1:
        ans += cnt[e.R].int64 * (N - cnt[e.R]).int64 * e.w.int64
        cnt[e.L] += cnt[e.R]
    else:
        ans += cnt[e.L].int64 * (N - cnt[e.L]).int64 * e.w.int64
        cnt[e.R] += cnt[e.L]
    cnte[e.R] -= 1
    cnte[e.L] -= 1
    if cnte[e.R] == 1:
        for t in nodes[e.R]:
            if used[t] == false:
                que.addLast(t)
    elif cnte[e.L] == 1:
        for t in nodes[e.L]:
            if used[t] == false:
                que.addLast(t)
    
echo ans * 2
0