結果

問題 No.1488 Max Score of the Tree
ユーザー 小野寺健小野寺健
提出日時 2021-04-24 16:20:32
言語 Java21
(openjdk 21)
結果
AC  
実行時間 267 ms / 2,000 ms
コード長 1,887 bytes
コンパイル時間 3,475 ms
コンパイル使用メモリ 78,128 KB
実行使用メモリ 106,808 KB
最終ジャッジ日時 2024-07-04 09:07:11
合計ジャッジ時間 10,816 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 252 ms
106,428 KB
testcase_01 AC 252 ms
89,388 KB
testcase_02 AC 262 ms
106,808 KB
testcase_03 AC 259 ms
106,376 KB
testcase_04 AC 267 ms
106,464 KB
testcase_05 AC 150 ms
54,304 KB
testcase_06 AC 176 ms
63,012 KB
testcase_07 AC 213 ms
75,748 KB
testcase_08 AC 193 ms
67,076 KB
testcase_09 AC 206 ms
65,688 KB
testcase_10 AC 203 ms
75,864 KB
testcase_11 AC 258 ms
106,412 KB
testcase_12 AC 134 ms
53,836 KB
testcase_13 AC 189 ms
59,324 KB
testcase_14 AC 200 ms
76,116 KB
testcase_15 AC 184 ms
65,768 KB
testcase_16 AC 142 ms
56,068 KB
testcase_17 AC 168 ms
63,116 KB
testcase_18 AC 225 ms
82,628 KB
testcase_19 AC 186 ms
66,180 KB
testcase_20 AC 169 ms
59,312 KB
testcase_21 AC 149 ms
56,504 KB
testcase_22 AC 181 ms
65,604 KB
testcase_23 AC 130 ms
54,244 KB
testcase_24 AC 128 ms
54,116 KB
testcase_25 AC 140 ms
53,852 KB
testcase_26 AC 169 ms
63,380 KB
testcase_27 AC 161 ms
56,692 KB
testcase_28 AC 164 ms
59,048 KB
testcase_29 AC 168 ms
59,172 KB
testcase_30 AC 227 ms
83,444 KB
testcase_31 AC 259 ms
106,516 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

package yukicoder;

import java.util.Scanner;
import java.util.List;
import java.util.ArrayList;

public class No1488 {
	
	private static class Peek {
		List<Integer> edge;
		List<Integer[]> peeks;
		public Peek() {
			edge = new ArrayList<Integer>();
			peeks = new ArrayList<Integer[]>();
		}
	}
	
	private static class Queue {
		int i;
		List<Integer> edge;
		public Queue(int i) {
			this.i = i;
			this.edge = new ArrayList<Integer>();
		}
		public Queue(int i, List<Integer> edge, int c) {
			this.i = i;
			this.edge = new ArrayList<Integer>(edge);
			this.edge.add(c);
		}
	}

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		int N = scan.nextInt();
		int K = scan.nextInt();
		List<Peek> v = new ArrayList<Peek>();
		for (int i=0; i < N; i++) {
			v.add(new Peek());
		}
		int[] weight = new int[N-1];
		for (int i=0; i < N-1; i++) {
			int a = scan.nextInt() - 1;
			int b = scan.nextInt() - 1;
			int c = scan.nextInt();
			v.get(a).peeks.add(new Integer[] {b, i});
			v.get(b).peeks.add(new Integer[] {a, i});
			weight[i] = c;
		}
		scan.close();
		List<Queue> q = new ArrayList<Queue>();
		q.add(new Queue(0));
		int[] e = new int[N-1];
		while (q.size() > 0) {
			Queue qi = q.remove(0);
			v.get(qi.i).edge = qi.edge;
			boolean cnt = true;
			for (Integer[] j : v.get(qi.i).peeks) {
				if (j[0] == 0 || v.get(j[0]).edge.size() > 0) {
					continue;
				}
				cnt = false;
				q.add(new Queue(j[0], qi.edge, j[1]));
			}
			if (cnt) {
				for (int j : qi.edge) {
					e[j]++;
				}
			}
		}
		int[][] dp = new int[N][K+1];
		int val = 0;
		for (int i=0; i < N-1; i++) {
			val += weight[i] * e[i];
			for (int w = 0; w <= K; w++) {
				if (w >= weight[i]) {
					dp[i+1][w] = Math.max(dp[i][w-weight[i]] + weight[i] * e[i], dp[i][w]);
				} else {
					dp[i+1][w] = dp[i][w];
				}
			}
		}
		System.out.println(dp[N-1][K]+val);
	}

}
0