tookunn’s diary

主に競技プログラミング関係

DISCO presents ディスカバリーチャンネル プログラミングコンテスト2016 本選 B DDPC特別ビュッフェⅡ

本戦で部分点取ってから何も手を付けてなかったので、解説見ながらやってみた。

考え

  • 求める最小の時刻を二分探索を使って、貪欲に求めていく。
  • ある時刻xに選択することができる最大の美味しさを持つ料理を取っていく。そして、その操作を時刻midから0までやっていく。


参考
Disco Presents ディスカバリーチャンネルプログラミングコンテスト2016 本選 解説
問題B : DDPC(DISCO presents ディスカバリーチャンネル プログラミングコンテスト)2016 本選 - selpo's diary


ソースコード

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.NoSuchElementException;
import java.util.PriorityQueue;
public class Main {
	int N,X;
	int[] T,A;

	public void solve() {
		N = nextInt();
		X = nextInt();

		T = new int[N];
		A = new int[N];
		int maxT = 0;


		for(int i = 0;i < N;i++){
			T[i] = nextInt();
			maxT = Math.max(maxT,T[i]);
		}

		for(int i = 0;i < N;i++){
			A[i] = nextInt();
		}

		ArrayList<Integer>[] list = new ArrayList[maxT + 1];

		for(int i = 0;i < maxT + 1;i++){
			list[i] = new ArrayList<Integer>();
		}

		//時刻T[i]に選ぶことが出来る料理の美味しさA[i]をまとめておく。
		for(int i = 0;i < N;i++){
			list[T[i]].add(A[i]);
		}

		int low = 0;
		int high = maxT + 1;

		while(high - low > 1){
			int mid = ((high + low) >> 1);
			PriorityQueue<Integer> pq = new PriorityQueue<Integer>();

			//時刻mid以下の場合には必ず選ぶことができる料理
			for(int i = mid + 1;i < maxT + 1;i++){
				for(int j = 0;j < list[i].size();j++){
					pq.add(-list[i].get(j));
				}
			}
			
			//美味しさの合計はintでは収まらない場合が存在するので、気を付けよう。
			long nowA = 0;
			for(int i = mid;i > 0;i--){

				//時刻iに選ぶことができる料理を追加する。
				for(int j = 0;j < list[i].size();j++){
					pq.add(-list[i].get(j));
				}

				if(pq.size() > 0){
					//PriorityQueueに-を掛けて追加したので、さらに-を掛けて元に戻す。
					nowA += -pq.poll();
				}
			}

			if(nowA >= X){
				high = mid;
			}else{
				low = mid;
			}
		}

		out.println(high == maxT + 1 ? - 1: high);
	}

	public static void main(String[] args) {
		out.flush();
		new Main().solve();
		out.close();
	}

	/* Input */
	private static final InputStream in = System.in;
	private static final PrintWriter out = new PrintWriter(System.out);
	private final byte[] buffer = new byte[2048];
	private int p = 0;
	private int buflen = 0;

	private boolean hasNextByte() {
		if (p < buflen)
			return true;
		p = 0;
		try {
			buflen = in.read(buffer);
		} catch (IOException e) {
			e.printStackTrace();
		}
		if (buflen <= 0)
			return false;
		return true;
	}

	public boolean hasNext() {
		while (hasNextByte() && !isPrint(buffer[p])) {
			p++;
		}
		return hasNextByte();
	}

	private boolean isPrint(int ch) {
		if (ch >= '!' && ch <= '~')
			return true;
		return false;
	}

	private int nextByte() {
		if (!hasNextByte())
			return -1;
		return buffer[p++];
	}

	public String next() {
		if (!hasNext())
			throw new NoSuchElementException();
		StringBuilder sb = new StringBuilder();
		int b = -1;
		while (isPrint((b = nextByte()))) {
			sb.appendCodePoint(b);
		}
		return sb.toString();
	}

	public int nextInt() {
		return Integer.parseInt(next());
	}

	public long nextLong() {
		return Long.parseLong(next());
	}

	public double nextDouble() {
		return Double.parseDouble(next());
	}
}