結果

問題 No.528 10^9と10^9+7と回文
ユーザー 37zigen
提出日時 2017-06-10 01:26:49
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 2,421 bytes
コンパイル時間 2,364 ms
コンパイル使用メモリ 88,300 KB
実行使用メモリ 73,716 KB
最終ジャッジ日時 2024-09-22 21:39:40
合計ジャッジ時間 9,159 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.PrintWriter;
import java.math.BigInteger;
import java.util.Arrays;
import java.util.Scanner;

class Main implements Runnable {
	public static void main(String[] args) {
		// new Thread(null, new Main(), "",
		// Runtime.getRuntime().maxMemory()).start();
		new Main().run();
	}

	final long MODULO1 = 1_000_000_000;
	final long MODULO2 = 1_000_000_000 + 7;

	public void run() {
		Scanner sc = new Scanner(System.in);
		char[] s = sc.next().toCharArray();
		int[] arr = new int[s.length];
		for (int i = 0; i < s.length; ++i) {
			arr[i] = (int) (s[i] - '0');
		}
		solve(arr);
	}

	void solve(int[] arr) {
		int len = arr.length;
		System.out.println(f(arr, -1, 0, len, MODULO1));
		System.out.println(f(arr, -1, 0, len, MODULO2));
	}

	// state==-1:初期状態
	// state==0:上位桁を無視
	// state==1:上位桁が確定済みかつ上位桁が上限未満
	// state==2:上位桁が確定かつ上位桁が上限ぴったり
	// [s,d)の範囲のみを考えればよい
	long f(int[] arr, int state, int s, int d, long MODULO) {
		long sum = 0;
		if (d - s < 1) {
			if (state == 0) {
				return 0;
			} else if (state == 1 || state == 2) {
				return 1;
			} else {
				throw new AssertionError();
			}
		}
		if (state == -1) {
			sum = (sum + f(arr, 0, s + 1, d, MODULO));
			for (int i = 1; i <= Math.min(arr[s], arr[d - 1]); ++i) {
				if (i == arr[s]) {
					sum = (sum + f(arr, 2, s + 1, d - 1, MODULO)) % MODULO;
				} else {
					sum = (sum + f(arr, 1, s + 1, d - 1, MODULO)) % MODULO;
				}
			}
			return sum;
		} else if (state == 0 || state == 1) {
			int len = d - s;
			if (state == 0) {
				sum = (sum + f(arr, 0, s + 1, d, MODULO)) % MODULO;
				sum = (sum + 9 * f(arr, 1, s + 1, d - 1, MODULO)) % MODULO;
				return sum;
			} else {
				return pow(10, (len + 1) / 2, MODULO);
			}
		} else if (state == 2) {
			for (int i = 0; i <= Math.min(arr[s], arr[d - 1]); ++i) {
				if (i == arr[s]) {
					sum = (sum + f(arr, 2, s + 1, d - 1, MODULO)) % MODULO;
				} else {
					sum = (sum + f(arr, 1, s + 1, d - 1, MODULO)) % MODULO;
				}
			}
			return sum;
		} else {
			throw new AssertionError();
		}

	}

	long pow(long a, int n, long MODULO) {
		long ret = 1;
		for (; n > 0; n >>= 1, a = (a * a) % MODULO) {
			if (n % 2 == 1) {
				ret = (ret * a) % MODULO;
			}
		}
		return ret;
	}

	static void tr(Object... objects) {
		System.out.println(Arrays.deepToString(objects));
	}

}
0