結果

問題 No.1594 Three Classes
ユーザー kinkincindykinkincindy
提出日時 2021-07-09 22:58:33
言語 Java21
(openjdk 21)
結果
TLE  
実行時間 -
コード長 2,165 bytes
コンパイル時間 3,154 ms
コンパイル使用メモリ 76,488 KB
実行使用メモリ 61,060 KB
最終ジャッジ日時 2023-09-14 10:31:57
合計ジャッジ時間 6,917 ms
ジャッジサーバーID
(参考情報)
judge15 / judge14
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 193 ms
61,060 KB
testcase_01 AC 118 ms
55,388 KB
testcase_02 AC 120 ms
55,724 KB
testcase_03 TLE -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.lang.reflect.Array;
import java.util.*;

class Main {
    static List<Set<Integer>> poss = new ArrayList<>();
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = Integer.parseInt(sc.next());
        long[] e = new long[n];
        long sum = 0;
        for (int i = 0; i < n; i++){
            e[i] = Integer.parseInt(sc.next());
            sum += e[i];
        }

        if (sum % 3 != 0){
            System.out.print("No");
            return;
        }
        long target = sum / 3;
        boolean[] visited = new boolean[n];
        boolean flag = false;
        dfs(0, n, e,visited, new HashSet<>(), target);
        for (int i = 0; i < poss.size(); i++){
            for (int j = i + 1; j < poss.size(); j++){
                for (int k = j + 1; k < poss.size(); k++){
                    Set<Integer> set1 = poss.get(i);
                    Set<Integer> set2 = poss.get(j);
                    Set<Integer> set3 = poss.get(k);
                    for (int a: set1){
                        if (!set2.contains(a) && !set3.contains(a)){
                            for (int b: set2){
                                if (!set1.contains(b) && !set3.contains(b)){
                                    flag = true;
                                    break;
                                }
                            }
                        }
                    }
                }
            }
        }

        if (flag){
            System.out.print("Yes");
        } else {
            System.out.print("No");
        }
    }

    private static void dfs(int index, int n, long[] e, boolean[] visited, Set<Integer> path, long remain){
        if (remain <= 0 || index > n){
            if (remain == 0){
                poss.add(new HashSet<>(path));
            }
            return;
        }
        for (int i = index; i < n; i++){
            dfs(i + 1, n, e, visited, path, remain);
            path.add(i);
            visited[i] = true;
            dfs(i + 1, n, e, visited, path, remain - e[i]);
            path.remove(i);
            visited[i] = false;
        }
    }
}
0