#319754 (C++11) No.834 Random Walk Trip

• 問題: No.834 Random Walk Trip
• ユーザー: leaf_1415
• 提出日時: 2019-02-25 21:15:35
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 19 ms / 2,000 ms
• コード長: 1,246 bytes
• コンパイル時間: 390 ms
• コンパイル使用メモリ: 53,500 KB
• 実行使用メモリ: 19,316 KB
• 最終ジャッジ日時: 2023-09-06 23:56:18
• 合計ジャッジ時間: 2,433 ms
• ジャッジサーバーID （参考情報）: judge14 / judge15

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	17 ms; 19,044 KB
testcase_01	AC	18 ms; 18,996 KB
testcase_02	AC	18 ms; 19,132 KB
testcase_03	AC	18 ms; 18,996 KB
testcase_04	AC	18 ms; 18,996 KB
testcase_05	AC	18 ms; 19,056 KB
testcase_06	AC	18 ms; 19,048 KB
testcase_07	AC	18 ms; 18,992 KB
testcase_08	AC	18 ms; 19,044 KB
testcase_09	AC	18 ms; 19,124 KB
testcase_10	AC	18 ms; 19,120 KB
testcase_11	AC	18 ms; 19,188 KB
testcase_12	AC	18 ms; 19,056 KB
testcase_13	AC	18 ms; 18,992 KB
testcase_14	AC	18 ms; 19,040 KB
testcase_15	AC	18 ms; 19,048 KB
testcase_16	AC	18 ms; 18,984 KB
testcase_17	AC	18 ms; 19,116 KB
testcase_18	AC	18 ms; 19,044 KB
testcase_19	AC	18 ms; 19,060 KB
testcase_20	AC	18 ms; 19,048 KB
testcase_21	AC	18 ms; 18,992 KB
testcase_22	AC	18 ms; 18,992 KB
testcase_23	AC	19 ms; 18,992 KB
testcase_24	AC	18 ms; 19,316 KB
testcase_25	AC	19 ms; 18,996 KB

ソースコード

#include <iostream>
#define llint long long
#define mod 1000000007

using namespace std;

llint fact[1000005], fact_inv[1000005];

llint modpow(llint a, llint n)
{
	if(n == 0) return 1;
	if(n % 2){
		return ((a%mod) * (modpow(a, n-1)%mod)) % mod;
	}
	else{
		return modpow((a*a)%mod, n/2) % mod;
	}
}

void make_fact()
{
	llint val = 1;
	fact[0] = 1;
	for(int i = 1; i < 1000005; i++){
		val *= i;
		val %= mod;
		fact[i] = val;
	}
	fact_inv[1000004] = modpow(fact[1000004], mod-2);
	for(int i = 1000003; i >= 0; i--){
		fact_inv[i] = fact_inv[i+1] * (i+1) % mod;
	}
}

llint comb(llint n, llint k)
{
	llint ret = 1;
	ret *= fact[n];
	ret *= fact_inv[k], ret %= mod;
	ret *= fact_inv[n-k], ret %= mod;
	return ret;
}

llint n, m;

int main(void)
{
	cin >> n >> m;
	make_fact();
	
	if(n == 1){
		cout << 1 << endl;
		return 0;
	}
	
	llint ans = 0;
	if(m%2){
		for(int i = 1; ; i++){
			if((m-1)/2+i*n > m) break;
			ans += comb(m, (m-1)/2+i*n), ans %= mod;
		}
		for(int i = 0; ; i++){
			if((m+1)/2+i*n > m) break;
			ans += comb(m, (m+1)/2+i*n), ans %= mod;
		}
	}
	else{
		for(int i = 1; ; i++){
			if(m/2+i*n > m) break;
			ans += 2 * comb(m, m/2+i*n), ans %= mod;
		}
		ans += comb(m, m/2), ans %= mod;
	}
	cout << ans << endl;
	
	return 0;
}