結果

問題 No.1594 Three Classes
ユーザー Kenneth YongKenneth Yong
提出日時 2021-07-09 22:52:15
言語 Kotlin
(2.1.0)
結果
AC  
実行時間 315 ms / 2,000 ms
コード長 787 bytes
コンパイル時間 12,590 ms
コンパイル使用メモリ 437,196 KB
実行使用メモリ 51,680 KB
最終ジャッジ日時 2024-11-16 08:21:36
合計ジャッジ時間 20,166 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

fun main() {
	val n = readLine()!!.toInt();
	val list = readLine()!!.split(" ").map {it -> it.toLong()};
	var sum = list.reduce {p, q -> p + q};
	if (sum % 3L != 0L) {
		println("No");
		return;
	}
	sum /= 3;
	
	for (i in 1 until 1.shl(n)) {
		var current = 0L;
		for (j in 0 until n) {
			if ((i and 1.shl(j)) > 0) {
				current += list[j];
			}
		}
		if (current == 2 * sum) {
			var mask = i;
			while (mask > 0) {
				current = 0L;
				for (j in 0 until n) {
					if ((mask and 1.shl(j)) > 0) {
						current += list[j];
					}
				}
				if (current == sum) {
					println("Yes");
					return;
				}
				
				mask = (mask - 1) and i;
			}
		}
	}
	
	println("No");
}
/*
C:\Users\kenne\OneDrive\Desktop\competitive_programming\kotlinmain.kt
*/
0