結果

問題 No.3660 LIS on Tree
コンテスト
ユーザー 👑 KA37RI
提出日時 2026-08-30 15:22:35
言語 Rust
(1.97.1 + proconio + num + itertools)
コンパイル:
/usr/bin/rustc_custom
実行:
./target/release/main
結果
AC  
実行時間 62 ms / 2,000 ms
+ 433µs
コード長 857 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 15,313 ms
コンパイル使用メモリ 186,448 KB
実行使用メモリ 21,536 KB
最終ジャッジ日時 2026-08-30 15:22:53
合計ジャッジ時間 15,523 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

use std::collections::VecDeque;

use proconio::{input, marker::Usize1};

fn solve() -> u64 {
  input! {
    n: usize,
    s: [u64; n],
    ab: [(Usize1, Usize1); n - 1],
  }
  let mut g = vec![vec![]; n];
  let mut d = vec![0; n];
  for &(ai, bi) in ab.iter() {
    if s[ai] > s[bi] {
      g[bi].push(ai);
      d[ai] += 1;
    } else if s[ai] < s[bi] {
      g[ai].push(bi);
      d[bi] += 1;
    }
  }
  let mut ls = vec![0; n];
  let mut q = VecDeque::new();
  d.iter().enumerate().for_each(|(i, &di)| if di == 0 {
    q.push_back(i);
    ls[i] = s[i];
  });
  while let Some(cur) = q.pop_front() {
    for &nxt in g[cur].iter() {
      ls[nxt] = ls[nxt].max(ls[cur] + s[nxt]);
      d[nxt] -= 1;
      if d[nxt] == 0 {
        q.push_back(nxt);
      }
    }
  }
  *ls.iter().max().unwrap()
}

fn main() {
  let ans = solve();
  println!("{}", ans);
}
0