結果

問題 No.15 カタログショッピング
ユーザー 37zigen37zigen
提出日時 2020-03-29 06:03:15
言語 Java21
(openjdk 21)
結果
AC  
実行時間 396 ms / 5,000 ms
コード長 3,984 bytes
コンパイル時間 2,311 ms
コンパイル使用メモリ 85,812 KB
実行使用メモリ 51,772 KB
最終ジャッジ日時 2024-06-10 18:10:44
合計ジャッジ時間 5,538 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 125 ms
41,912 KB
testcase_01 AC 125 ms
42,332 KB
testcase_02 AC 168 ms
42,864 KB
testcase_03 AC 140 ms
42,784 KB
testcase_04 AC 146 ms
42,752 KB
testcase_05 AC 333 ms
51,172 KB
testcase_06 AC 357 ms
51,412 KB
testcase_07 AC 369 ms
50,812 KB
testcase_08 AC 368 ms
51,492 KB
testcase_09 AC 396 ms
51,772 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.NoSuchElementException;
import java.util.Scanner;

public class Main {	
	public static void main(String[] args) {
		//long t = System.currentTimeMillis();
		new Main().run();
		//System.err.println(System.currentTimeMillis() - t);
	}
	
	long flip(long v,int n) {
		for(int i=0;i<n;++i) {
			int j=n-1-i;
			if(i>=j)continue;
			v^=((v>>i)&1)<<j;
			v^=((v>>j)&1)<<i;
			v^=((v>>i)&1)<<j;
		}
		return v;
	}
	
	void run() {
		Scanner sc=new Scanner(System.in);
		int N=sc.nextInt();
		long target=sc.nextLong();
		long[] P=new long[N];
		for(int i=0;i<N;++i)P[i]=sc.nextLong();
		int a=(N+1)/2;int b=N/2;
		long[][] s1=new long[1<<a][2];
		long[][] s2=new long[1<<b][2];
		for(int s=0;s<1<<a;++s) {
			s1[s][1]=s;
			for(int i=0;i<a;++i) {
				if((s>>i)%2==0)continue;
				s1[s][0]+=P[i];
			}
		}
		for(int s=0;s<1<<b;++s) {
			s2[s][1]=s;
			for(int i=0;i<b;++i) {
				if((s>>i)%2==0)continue;
				s2[s][0]+=P[i+a];
			}
		}
		Comparator<long[]> comp=new Comparator<long[]>() {
			@Override
			public int compare(long[] o1, long[] o2) {
				return Long.compare(o1[0], o2[0]);
			}
		};
		Arrays.sort(s1,comp);
		Arrays.sort(s2,comp);
		ArrayList<Long> ans=new ArrayList<>();
		for(int i=0;i<s1.length;++i) {
			int ok=s2.length-1;
			int ng=-1;
			while(ok-ng>1) {
				int middle=(ok+ng)/2;
				if(s2[middle][0]>=target-s1[i][0])ok=middle;
				else ng=middle;
			}
			if(s2[ok][0]!=target-s1[i][0])continue;
			while(ok<s2.length&&s2[ok][0]==target-s1[i][0]) {
				ans.add(flip((long)s2[ok++][1]<<a|s1[i][1],N));
			}
		}
		Collections.sort(ans);
		for(int i=ans.size()-1;i>=0;--i) {
			long s=flip(ans.get(i),N);
			for(int j=0;j<N;++j) {
				if((s>>j)%2==1) {
					System.out.print((j+1)+((s>>(j+1))==0?"\n":" "));
				}
			}
		}
		
	}

	
	
	static void tr(Object... objects) {
		System.out.println(Arrays.deepToString(objects));
	}

}
	
	
class FastScanner {
    private final InputStream in = System.in;
    private final byte[] buffer = new byte[1024];
    private int ptr = 0;
    private int buflen = 0;
    private boolean hasNextByte() {
        if (ptr < buflen) {
            return true;
        }else{
            ptr = 0;
            try {
                buflen = in.read(buffer);
            } catch (IOException e) {
                e.printStackTrace();
            }
            if (buflen <= 0) {
                return false;
            }
        }
        return true;
    }
    private int readByte() { if (hasNextByte()) return buffer[ptr++]; else return -1;}
    private static boolean isPrintableChar(int c) { return 33 <= c && c <= 126;}
    private void skipUnprintable() { while(hasNextByte() && !isPrintableChar(buffer[ptr])) ptr++;}
    public boolean hasNext() { skipUnprintable(); return hasNextByte();}
    public String next() {
        if (!hasNext()) throw new NoSuchElementException();
        StringBuilder sb = new StringBuilder();
        int b = readByte();
        while(isPrintableChar(b)) {
            sb.appendCodePoint(b);
            b = readByte();
        }
        return sb.toString();
    }
    public long nextLong() {
        if (!hasNext()) throw new NoSuchElementException();
        long n = 0;
        boolean minus = false;
        int b = readByte();
        if (b == '-') {
            minus = true;
            b = readByte();
        }
        if (b < '0' || '9' < b) {
            throw new NumberFormatException();
        }
        while(true){
            if ('0' <= b && b <= '9') {
                n *= 10;
                n += b - '0';
            }else if(b == -1 || !isPrintableChar(b)){
                return minus ? -n : n;
            }else{
                throw new NumberFormatException();
            }
            b = readByte();
        }
    }
    
    public int nextInt() {
    	return (int)nextLong();
    }
}
0