CODE FESTIVAL 2016 qual B D
考察
本番中は貪欲っぽいな思って、色々考察して試して時間切れを迎えた。
どういう発想で解法に近づくのかを記録しておく。
・問題概要から数列Aのある要素はその要素の前の要素によって操作が変わる?だから前から走査していく?
・DPにするにしても遷移が決まっているので違う?じゃあ貪欲?
・なので単純に前の要素から貪欲に走査してみる。
・まず最初の要素から最大の操作回数を得るにはA[0] - 1回の操作を行う必要がある。
・次の2番目の要素では最初の要素は1になっているはずなので、Pを1にすることはできない。なので、P=2で引いていくことにしてみる。
・もし2番目の要素がP以下だったら引くことができない(0になるため)。P = A[i]の場合、次の要素からはP =A[i] + 1にしていかないと2番目の要素で所持金が0になるまで買ってしまう。
・Pより大きかったら1以上P以下になるまで引いてみる。
・これを最後の要素までやっていく。
ソースコード
import java.io.IOException; import java.io.InputStream; import java.io.PrintWriter; import java.util.NoSuchElementException; public class Main { int N; int[] A; public void solve() { N = nextInt(); A = new int[N]; for(int i = 0;i < N;i++){ A[i] = nextInt(); } long max = 1L; long ans = (A[0] - 1);//最初の要素を1にし、A[0] - 1をカウント for(int i = 1;i < N;i++){ if(A[i] > max + 1){ //A[i] - 1は1は残さないといけないため //1以上を残してmax+1ずつ減らしていく ans += (A[i] - 1) / (max + 1); }else if(A[i] == max + 1){ //A[i]はmax+1で減らすと0になってしまうので、スルー //次の要素からはmax = A[i] + 1にしないといけない max++; } } out.println(ans); } 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()); } }
どうせまた解き直す時、この問題の発想忘れてるんだろうな~