結果

問題 No.15 カタログショッピング
コンテスト
ユーザー uafr_cs
提出日時 2015-10-12 01:18:33
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 1,941 bytes
コンパイル時間 2,051 ms
コンパイル使用メモリ 82,880 KB
実行使用メモリ 48,060 KB
最終ジャッジ日時 2024-07-21 07:07:00
合計ジャッジ時間 6,333 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 6 WA * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Arrays;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;

public class Main {
	
	public static class Item implements Comparable<Item> {
		long cost;
		int index;
		
		public Item(long cost, int index) {
			super();
			this.cost = cost;
			this.index = index;
		}

		@Override
		public int compareTo(Item o) {
			if(Long.compare(o.cost, this.cost) != 0){
				return Long.compare(o.cost, this.cost);
			}else{
				return Integer.compare(this.index, o.index);
			}
		}
	}
	
	public static void dfs(final long S, final int N, final int deep, Item[] items, long[] maximum, LinkedList<Integer> stack, TreeSet<String> answers){
		if(S == 0){
			StringBuilder sb = new StringBuilder();
			boolean first = true;
			
			
			for(final int item : new TreeSet<Integer>(stack)){
				if(first){
					first = false;
				}else{
					sb.append(" ");
				}
				
				sb.append(item + 1);
			}
			answers.add(sb.toString());
			
			return;
		}else if(deep >= N){
			return;
		}else if(maximum[deep] < S){
			return;
		}
		
		if(S >= items[deep].cost){
			stack.addLast(items[deep].index);
			dfs(S - items[deep].cost, N, deep + 1, items, maximum, stack, answers);
			stack.removeLast();
		}
		dfs(S, N, deep + 1, items, maximum, stack, answers);
	}
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		
		final int N = sc.nextInt();
		final long S = sc.nextLong();
		
		Item[] items = new Item[N];
		for(int i = 0; i < N; i++){
			items[i] = new Item(sc.nextLong(), i);
		}
		Arrays.sort(items);
		
		long[] maximum = new long[N];
		for(int i = N - 1; i >= 0; i--){
			maximum[i] = items[i].cost + (i == N - 1 ? 0 : maximum[i + 1]);
		}
		
		TreeSet<String> answers = new TreeSet<String>();
		dfs(S, N, 0, items, maximum, new LinkedList<Integer>(), answers);
	
		for(final String ans : answers){
			System.out.println(ans);
		}
	}

}
0