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()); } }