tookunn’s diary

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

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

どうせまた解き直す時、この問題の発想忘れてるんだろうな~