tookunn’s diary

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

RUPC2016 Day1 D スキャナー / Scanner

考察

当初、貪欲に時間のかかる紙から選んで、早くスキャンが終わっているスキャナーに振り分けた。が、通らなかった。

DP[i番目の紙][1つ目のスキャナーのスキャン終了時間][2つ目のスキャナーのスキャン終了時間] = このケースが存在するか(true または false)

3つ目のスキャナーのスキャン終了時間は、残り2つのスキャナーのスキャン終了時間が分かれば分かる。(この考察は解説見る前に気付いたが、ここからどうすればよいのか分からなかった。(メモ))

JavaDP[N+1][2500][2500]でギリギリ1.88secぐらいだったので以下記事のように2500 \to 900にした方が良い。

参考にさせていただいた解説記事:
pakapa104.hatenablog.com

ソースコード

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.NoSuchElementException;

public class Main {
	int N;
	int[] T;
	boolean[][][] dp;

	public void solve() {
		N = nextInt();
		T = new int[N];
		int sum = 0;
		for(int i = 0;i < N;i++){
			T[i] = nextInt();
			sum += T[i];
		}

		dp = new boolean[N + 1][2500 + 1][2500 + 1];
		dp[0][0][0] = true;

		for(int i = 0;i < N;i++){

			for(int j = 0;j < 2500 + 1;j++){

				for(int k = 0;k < 2500 + 1;k++){

					if(!dp[i][j][k])continue;

					if(j + T[i] < 2500)dp[i + 1][j + T[i]][k] = true;

					if(k + T[i] < 2500)dp[i + 1][j][k + T[i]] = true;

					dp[i + 1][j][k] = true;

				}

			}

		}

		int ans = Integer.MAX_VALUE;

		for(int i = 0;i < 2500 + 1;i++){
			for(int j = 0;j < 2500 + 1;j++){
				if(dp[N][i][j]){
					ans = Math.min(ans, Math.max(sum - i - j,Math.max(i, j)));
				}

			}
		}

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