結果

問題 No.834 Random Walk Trip
ユーザー 37zigen
提出日時 2019-05-25 01:00:50
言語 Java
(openjdk 23)
結果
AC  
実行時間 249 ms / 2,000 ms
コード長 1,712 bytes
コンパイル時間 1,856 ms
コンパイル使用メモリ 78,164 KB
実行使用メモリ 106,016 KB
最終ジャッジ日時 2024-09-17 14:59:31
合計ジャッジ時間 8,812 ms
ジャッジサーバーID
(参考情報)
judge6 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.util.*;
import java.io.*;

public class Main {
	public static void main(String[] args) throws Exception {
		new Main().run();
	}

	final long MOD = (long) 1e9 + 7;

	long[] fac = new long[2000000];
	long[] ifac = new long[2000000];
	long[] inv = new long[2000000];
	{
		fac[0] = ifac[0] = inv[0] = fac[1] = ifac[1] = inv[1] = 1;
		for (int i = 2; i < fac.length; ++i) {
			fac[i] = fac[i - 1] * i % MOD;
		}
		// 0 = p/i*i + (p%i)
		for (int i = 2; i < inv.length; ++i) {
			inv[i] = MOD - inv[(int) (MOD % i)] * (MOD / i) % MOD;
		}
		for (int i = 2; i < ifac.length; ++i) {
			ifac[i] = inv[i] * ifac[i - 1] % MOD;
		}
	}

	long comb(int n, int k) {
		return fac[n] * ifac[k] % MOD * ifac[n - k] % MOD;
	}

	void run() throws Exception {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int M = sc.nextInt();
		if (N == 1) {
			System.out.println(1);
			return;
		}
		long ans = 0;
		int[] a = new int[M + 2];
		a[0] = 1;
		for (int i = 1; i <= N; ++i) {
			for (int j = 0; i + j < a.length; j += 2 * N) {
				a[i + j] = i;
				if (i + j + N < a.length)
					a[i + j + N] = N + 1 - i;
			}
		}
		for (int i = 1; i < a.length; ++i) {
			if (a[i] != 1)
				continue;
			int r = i - 1;
			if (r % 2 != M % 2)
				continue;
			if (M - r < 0)
				continue;
			int l = (M - r) / 2;
			r += l;
			ans = (ans + comb(l + r, l)) % MOD;
		}
		for (int i = 1; i < a.length; ++i) {
			if (a[i] != 1)
				continue;
			int r = i;
			if (r % 2 != M % 2)
				continue;
			if (M - r < 0)
				continue;
			int l = (M - r) / 2;
			r += l;
			ans = (ans + comb(l + r, l)) % MOD;
		}
		System.out.println(ans);
	}

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

}
0