#1025290 (Java21) No.2957 Combo Deck Builder

• 問題: No.2957 Combo Deck Builder
• ユーザー: ks2m
• 提出日時: 2024-11-08 22:28:31
• 言語: Java21 (openjdk 21)
• 結果: TLE
• 実行時間: -
• コード長: 1,711 bytes
• コンパイル時間: 2,713 ms
• コンパイル使用メモリ: 81,660 KB
• 実行使用メモリ: 79,820 KB
• 最終ジャッジ日時: 2024-11-08 22:29:14
• 合計ジャッジ時間: 40,671 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	60 ms; 37,020 KB
testcase_01	AC	65 ms; 37,552 KB
testcase_02	AC	62 ms; 37,504 KB
testcase_03	AC	60 ms; 37,160 KB
testcase_04	AC	61 ms; 37,204 KB
testcase_05	AC	60 ms; 37,344 KB
testcase_06	TLE	-
testcase_07	TLE	-
testcase_08	TLE	-
testcase_09	TLE	-
testcase_10	TLE	-
testcase_11	TLE	-
testcase_12	TLE	-
testcase_13	TLE	-
testcase_14	TLE	-
testcase_15	TLE	-
testcase_16	TLE	-
testcase_17	TLE	-
testcase_18	TLE	-
testcase_19	TLE	-
testcase_20	TLE	-
testcase_21	TLE	-
testcase_22	TLE	-
testcase_23	TLE	-
testcase_24	TLE	-
testcase_25	TLE	-
testcase_26	TLE	-
testcase_27	TLE	-
testcase_28	TLE	-
testcase_29	TLE	-
testcase_30	TLE	-
testcase_31	AC	931 ms; 65,060 KB
testcase_32	TLE	-
testcase_33	AC	927 ms; 68,440 KB
testcase_34	TLE	-
testcase_35	TLE	-
testcase_36	AC	62 ms; 37,332 KB
testcase_37	AC	63 ms; 37,516 KB
testcase_38	AC	63 ms; 37,260 KB
testcase_39	AC	61 ms; 37,564 KB

ソースコード

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.PriorityQueue;
import java.util.TreeSet;

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		PriorityQueue<Obj> que1 = new PriorityQueue<>((o1, o2) -> o1.v - o2.v);
		PriorityQueue<Obj> que2 = new PriorityQueue<>((o1, o2) -> o2.v - o1.v);
		for (int i = 0; i < n; i++) {
			String[] sa = br.readLine().split(" ");
			Obj o = new Obj();
			o.c = Integer.parseInt(sa[0]);
			o.x = Integer.parseInt(sa[1]);
			o.y = Integer.parseInt(sa[2]);
			o.v = o.x - o.y;
			if (o.v < 0) {
				que1.add(o);
			} else {
				que2.add(o);
			}
		}
		br.close();

		TreeSet<Integer> set = new TreeSet<>();
		for (int i = 0; i < n; i++) {
			set.add(i);
		}
		Obj[] arr = new Obj[n];
		while (!que1.isEmpty()) {
			Obj o = que1.poll();
			Integer k = set.lower(o.c);
			if (k == null) {
				k = set.last();
			}
			arr[k] = o;
			set.remove(k);
		}

		int l = 0;
		for (int r = 0; r < n; r++) {
			if (arr[r] != null) {
				if (l < r) {
					arr[l] = arr[r];
					arr[r] = null;
				}
				l++;
			}
		}

		set.clear();
		for (int i = n - 1; i >= 0; i--) {
			if (arr[i] != null) {
				break;
			}
			set.add(i);
		}

		while (!que2.isEmpty()) {
			Obj o = que2.poll();
			Integer k = set.ceiling(o.c);
			if (k == null) {
				k = set.first();
			}
			arr[k] = o;
			set.remove(k);
		}

		long ans = 0;
		for (int i = 0; i < n; i++) {
			Obj o = arr[i];
			if (o.c <= i) {
				ans += o.x;
			} else {
				ans += o.y;
			}
		}
		System.out.println(ans);
	}

	static class Obj {
		int c, x, y, v;
	}
}