結果

問題 No.15 カタログショッピング
コンテスト
ユーザー jp_ste
提出日時 2016-05-18 02:13:35
言語 Java
(openjdk 23)
結果
AC  
実行時間 375 ms / 5,000 ms
コード長 3,838 bytes
コンパイル時間 2,296 ms
コンパイル使用メモリ 89,888 KB
実行使用メモリ 59,932 KB
最終ジャッジ日時 2024-10-06 05:24:04
合計ジャッジ時間 5,503 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.Scanner;

public class Main {
    static int n;
    static long s;
    static long[] list;
    
    public static void main(String[] args) {
        try (Scanner scan = new Scanner(System.in)) {
            n = Integer.parseInt(scan.next());
            s = Long.parseLong(scan.next());
            list = new long[n];
            for(int i=0; i<n; i++) {
                list[i] = Long.parseLong(scan.next());
            }
            
            ArrayList<Integer> indexList = new ArrayList<Integer>();
            ArrayList<Node> left = new ArrayList<Node>();
            ArrayList<Node> right = new ArrayList<Node>();
            ArrayList<String> outList = new ArrayList<String>();
            
            solve(0, n/2, 0, left, indexList);
            solve(n/2, n, 0, right, indexList);
            Collections.sort(right);
            
            for(int i=0; i<left.size(); i++) {
                int l = 0;
                int r = right.size()-1;
                while(l <= r) {
                    int m = (l+r+1)/2;
                    long v = left.get(i).v + right.get(m).v;
                    if(v == s) {
                        outList.add((left.get(i).add(right.get(m)).toString()));
                        long now = right.get(m).v;
                        while(--m >= 0) {
                            if(now != right.get(m).v) break;
                            outList.add((left.get(i).add(right.get(m)).toString()));
                        }
                        break;
                    } else if(v > s) {
                        r = m-1;
                    } else if(v < s) {
                        l = m+1;
                    }
                }
            }
            
            Collections.sort(outList, new UserComparator());
            for(String str : outList) {
                System.out.println(str);
            }
        }
    }
    
    static void solve(int i, int n, long sum, ArrayList<Node> nowList, ArrayList<Integer> indexList) {
        if(i == n) {
            nowList.add(new Node(sum, (ArrayList<Integer>)indexList.clone()));
            return;
        }
        solve(i+1, n, sum, nowList, indexList);
        indexList.add(i);
        solve(i+1, n, sum + list[i], nowList, indexList);
        indexList.remove(Integer.valueOf(i));
    }
    
    static class Node implements Comparable<Node>{
        long v;
        ArrayList<Integer> indexes;
        Node(long v, ArrayList<Integer> indexes) {
            this.v = v;
            this.indexes = indexes;
        }
        @Override
        public int compareTo(Node o) {
            return (int)(v - o.v);
        }
        public Node add(Node o) {
            ArrayList<Integer> ret = new ArrayList<Integer>();
            ret.addAll(indexes);
            ret.addAll(o.indexes);
            return new Node(v, ret);
        }
        
        @Override
        public String toString() {
            StringBuffer sb = new StringBuffer();
            for(int i=0; i<indexes.size(); i++) {
                if(i>0) sb.append(" ");
                sb.append(indexes.get(i)+1);
            }
            return sb.toString();
        }
    }
    
    final static class UserComparator implements Comparator {
        public int compare(Object o1, Object o2) {
            String values1[] = ((String)o1).split(" ");
            String values2[] = ((String)o2).split(" ");
            
            int i=0;
            while(true) {
                long v1 = Long.parseLong(values1[i]);
                long v2 = Long.parseLong(values2[i]);
                if(v1 == v2) {
                    i++;
                    continue;
                }
                return (int)(v1 - v2);
            }
        }
    }
}
0