読者です 読者をやめる 読者になる 読者になる

tookunn’s diary

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

第3回 ドワンゴからの挑戦状 予選 B ニコニコレベル

考察

文字列を左から順に見ていき,'2','5','?'の時で場合分けして状態を遷移する。

2の場合
  • i番目の文字が2である状態からi+1番目の文字が5である状態に遷移する
5の場合
  • i番目の文字が5である状態からi+1番目の文字が2である状態に遷移する
  • しかし,ニコニコ文字列は252525..のように2から始まる文字列なので,i-1番目の文字が2である必要がある
?の場合
  • 2,5の場合で行う遷移を両方行えばよい

dp[i][j] = i番目の文字がjであり,i番目の文字まで見た時の連続した最長のニコニコ文字列の長さ


書いてる途中で気付いたけど,dp[N][10]じゃなくてdp[N][2]で良さそうな気がする(2,5の2通りしか考慮しないため)

ソースコード

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

public class Main {
	String T;
	char[] ch;
	int N;
	int[][] dp;

	public void solve() {
		T = next();
		ch = T.toCharArray();
		N = T.length();
		dp = new int[N+1][10];
		//dp[i番目][i番目の文字]
		for(int i = 0;i < N;i++){
			if(ch[i] == '?'){
				if(dp[i][5]%2==1)dp[i + 1][2] = Math.max(dp[i+1][2],dp[i][5]+1);
				dp[i + 1][5] = Math.max(dp[i+1][5],dp[i][2]+1);
			}else if(ch[i] == '2'){
				dp[i + 1][5] = Math.max(dp[i+1][5],dp[i][2]+1);
			}else if(ch[i] == '5'){
				if(dp[i][5]%2==1)dp[i + 1][2] = Math.max(dp[i+1][2],dp[i][5]+1);
			}
		}

		int ans = 0;
		for(int i = 0;i < N+1;i++){
			for(int j = 0;j < 10;j++){
				ans = Math.max(ans,dp[i][j]);
			}
		}
		out.println(ans/2 * 2);
	}

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