結果

問題 No.1594 Three Classes
ユーザー merlinmerlin
提出日時 2021-07-09 21:38:36
言語 Java21
(openjdk 21)
結果
AC  
実行時間 63 ms / 2,000 ms
コード長 1,597 bytes
コンパイル時間 3,253 ms
コンパイル使用メモリ 78,724 KB
実行使用メモリ 37,164 KB
最終ジャッジ日時 2024-11-16 07:36:35
合計ジャッジ時間 4,217 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 63 ms
36,840 KB
testcase_01 AC 53 ms
36,552 KB
testcase_02 AC 53 ms
36,680 KB
testcase_03 AC 51 ms
36,388 KB
testcase_04 AC 51 ms
36,624 KB
testcase_05 AC 63 ms
37,072 KB
testcase_06 AC 52 ms
36,668 KB
testcase_07 AC 53 ms
36,736 KB
testcase_08 AC 52 ms
36,668 KB
testcase_09 AC 63 ms
37,128 KB
testcase_10 AC 61 ms
37,164 KB
testcase_11 AC 63 ms
37,164 KB
testcase_12 AC 62 ms
36,912 KB
testcase_13 AC 51 ms
36,616 KB
testcase_14 AC 51 ms
36,508 KB
testcase_15 AC 54 ms
36,860 KB
testcase_16 AC 53 ms
36,824 KB
testcase_17 AC 53 ms
36,620 KB
testcase_18 AC 53 ms
36,752 KB
testcase_19 AC 52 ms
36,404 KB
testcase_20 AC 52 ms
36,740 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.*;
import java.util.*;

class Main
{
    public static void main(String args[])throws Exception
    {
        BufferedReader bu=new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb=new StringBuilder();
        int n=Integer.parseInt(bu.readLine());
        int i,sum=0,ar[]=new int[n];
        ArrayList<Integer> a=new ArrayList<>();
        String s[]=bu.readLine().split(" ");
        for(i=0;i<n;i++)
        {
            ar[i]=Integer.parseInt(s[i]);
            a.add(i);
            sum+=ar[i];
        }
        if(sum%3!=0) {System.out.println("No"); return;}

        sum/=3;
        a=subset(a,ar,sum);
        boolean vis[]=new boolean[n];
        int sua=0,sub=0;
        for(int x:a)
        {
            vis[x]=true;
            sua+=ar[x];
        }

        ArrayList<Integer> b=new ArrayList<>();
        for(i=0;i<n;i++)
        if(!vis[i]) b.add(i);

        b=subset(b,ar,sum);
        for(int x:b) sub+=ar[x];
        if(sua==sub && sua==sum) sb.append("Yes\n");
        else sb.append("No\n");
        System.out.print(sb);
    }

    static ArrayList<Integer> subset(ArrayList<Integer> a,int ar[],int sum)
    {
        ArrayList<Integer> ans=new ArrayList<>();
        int i,n=a.size(),m=1<<n,j;
        for(i=0;i<m;i++)
        {
            int s=0;
            for(j=0;j<n;j++)
            if(((i>>j)&1)==1) s+=ar[a.get(j)];
            if(s==sum)
            {
                for(j=0;j<n;j++)
                if(((i>>j)&1)==1) ans.add(a.get(j));
                return ans;
            }
        }
        return ans;
    }
}
0