結果

問題 No.1488 Max Score of the Tree
コンテスト
ユーザー dot_haraai
提出日時 2021-05-31 10:45:06
言語 Nim
(2.2.0)
結果
AC  
実行時間 158 ms / 2,000 ms
コード長 1,745 bytes
コンパイル時間 3,744 ms
コンパイル使用メモリ 87,552 KB
実行使用メモリ 81,536 KB
最終ジャッジ日時 2024-11-08 21:08:29
合計ジャッジ時間 6,826 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
/home/judge/data/code/Main.nim(2, 18) Warning: Use the new 'sugar' module instead; future is deprecated [Deprecated]
/home/judge/data/code/Main.nim(1, 8) Warning: imported and not used: 'times' [UnusedImport]
/home/judge/data/code/Main.nim(2, 26) Warning: imported and not used: 'strformat' [UnusedImport]
/home/judge/data/code/Main.nim(2, 18) Warning: imported and not used: 'future' [UnusedImport]
/home/judge/data/code/Main.nim(2, 37) Warning: imported and not used: 'deques' [UnusedImport]
/home/judge/data/code/Main.nim(1, 41) Warning: imported and not used: 'algorithm' [UnusedImport]
/home/judge/data/code/Main.nim(1, 52) Warning: imported and not used: 'tables' [UnusedImport]
/home/judge/data/code/Main.nim(1, 60) Warning: imported and not used: 'sets' [UnusedImport]
/home/judge/data/code/Main.nim(1, 66) Warning: imported and not used: 'lists' [UnusedImport]
/home/judge/data/code/Main.nim(1, 73) Warning: imported and not used: 'intsets' [UnusedImport]
/home/judge/data/code/Main.nim(2, 8) Warning: imported and not used: 'critbits' [UnusedImport]

ソースコード

diff # 

import times, strutils, sequtils, math, algorithm, tables, sets, lists, intsets
import critbits, future, strformat, deques
template `max=`(x,y) = x = max(x,y)
template `min=`(x,y) = x = min(x,y)
template `mod=`(x,y) = x = x mod y
template scan2 = (scan(), scan())
template scan3 = (scan(), scan())
let read* = iterator: string {.closure.} =
    while true: (for s in stdin.readLine.split: yield s)
proc scan(): int = read().parseInt
proc scanf(): float = read().parseFloat
proc toInt(c:char): int =
    return int(c) - int('0')



proc solve()=
  var
    n = scan()
    k = scan()
    es = newseqwith(n,newseq[int]())
    cost = newseqwith(n,newseq[int]())
    esu = newseqwith(n,newseq[int]())
  for i in 0..<n-1:
    var
      (a,b,c)=(scan()-1,scan()-1,scan())
    es[a].add(b)
    es[b].add(a)

    cost[a].add(c)
    cost[b].add(c)
    esu[a].add(0)
    esu[b].add(0)
  proc dfs(par,p:int):int=
    if es[p].len==1 and es[p][0]==par:
      return 1
    for idx, nxt in es[p]:
      if nxt == par:continue
      var t = dfs(p,nxt)
      esu[p][idx]=t
      result+=t
  discard dfs(-1,0)
  # 各辺の使用回数を計算
  var
    edge = newseq[int]()
    score = newseq[int]()
    base = 0
  for p in 0..<n:
    for (c,m) in zip(cost[p],esu[p]):
      if m!=0:
        base+=c*m
        edge.add(c)
        score.add(m*c)
  var
    m = edge.len
    dp = newseqwith(m+1,newseqwith(k+1,0))
  # dp[i][j] := i個目の辺まで見たときに選んだ辺の総和がjであるときに得られるスコアの最大値
  for i in 1..m:
    for j in 0..k:
      if j - edge[i-1]>=0:
        dp[i][j] = max(dp[i-1][j-edge[i-1]]+score[i-1],dp[i-1][j])
      else:
        dp[i][j].max= dp[i-1][j]
  
  echo dp[m].max() + base


  

    


solve()
0