結果
問題 | No.1594 Three Classes |
ユーザー |
![]() |
提出日時 | 2021-07-09 21:38:36 |
言語 | Java (openjdk 23) |
結果 |
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 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
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; } }