結果

問題 No.1127 変形パスカルの三角形
ユーザー Kome_soudouKome_soudou
提出日時 2022-11-10 02:04:37
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 134 ms / 1,500 ms
コード長 1,566 bytes
コンパイル時間 1,744 ms
コンパイル使用メモリ 170,032 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2024-07-23 06:48:09
合計ジャッジ時間 5,276 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 134 ms
6,400 KB
testcase_02 AC 93 ms
5,504 KB
testcase_03 AC 69 ms
5,376 KB
testcase_04 AC 18 ms
5,376 KB
testcase_05 AC 63 ms
5,376 KB
testcase_06 AC 133 ms
6,272 KB
testcase_07 AC 40 ms
5,376 KB
testcase_08 AC 38 ms
5,376 KB
testcase_09 AC 110 ms
5,760 KB
testcase_10 AC 112 ms
5,888 KB
testcase_11 AC 93 ms
5,376 KB
testcase_12 AC 88 ms
5,376 KB
testcase_13 AC 83 ms
5,376 KB
testcase_14 AC 66 ms
5,376 KB
testcase_15 AC 35 ms
5,376 KB
testcase_16 AC 74 ms
5,376 KB
testcase_17 AC 38 ms
5,376 KB
testcase_18 AC 79 ms
5,376 KB
testcase_19 AC 96 ms
5,376 KB
testcase_20 AC 98 ms
5,632 KB
testcase_21 AC 99 ms
5,504 KB
testcase_22 AC 37 ms
5,376 KB
testcase_23 AC 111 ms
5,760 KB
testcase_24 AC 70 ms
5,376 KB
testcase_25 AC 75 ms
5,376 KB
testcase_26 AC 81 ms
5,376 KB
testcase_27 AC 95 ms
5,376 KB
testcase_28 AC 92 ms
5,376 KB
testcase_29 AC 34 ms
5,376 KB
testcase_30 AC 24 ms
5,376 KB
testcase_31 AC 100 ms
5,504 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

// a^b mod p O(logP)
int64_t modpow(int64_t a, int64_t b, int64_t p)
{
	int64_t ans = 1;
	while(b > 0)
	{
		if(b & 1) ans = ans * a % p; //bの最下位bitが1ならa^(2^i)をかける
		a = a * a % p;
		b >>= 1; //bを1bit右にシフト
	}
	return ans;
}
vector<int64_t> fact; // 要素数(N + 1)
vector<int64_t> fact_inv; // 要素数(N + 1)
// a! mod pと(a!)^-1 mod pを配列に記録 O(NlogP) (やや遅い まだ高速化可能)
// ModComb実行前にこれを実行すること
void combinit(int64_t n, int64_t p)
{
	fact.resize(n + 1);
	fact_inv.resize(n + 1);
	fact.at(0) = 1; fact.at(1) = 1;
	for(int64_t i = 2; i <= n; i++)
	{
		fact.at(i) = fact.at(i - 1) * i;
		fact.at(i) %= p;
	}
	fact_inv.at(0) = 1; fact_inv.at(1) = 1;
	for(int64_t i = 2; i <= n; i++) fact_inv.at(i) = modpow(fact.at(i), p - 2, p);
}
// nCk mod p O(1)
int64_t modcomb(int64_t n, int64_t k, int64_t p)
{
  return ((fact.at(n) * fact_inv.at(k)) % p * fact_inv.at(n - k)) % p;
}

int main()
{
	int64_t a, b, n, k;
	cin >> a >> b >> n >> k;
	const int64_t mod = 1000000007;
	a %= mod;
	b %= mod;
	combinit(n - 1, mod);
	int64_t x = ((modcomb(n - 1, k - 1, mod) * a) % mod + modcomb(n - 1, k - 2, mod) * b) % mod;
	cout << x << endl;
	
	int64_t y = 0;
	int64_t cnt;
	for(int64_t i = 0; i <= n; i++)
	{
		cnt = 0;
		if(i != n)
		{
			cnt += modcomb(n - 1, i, mod) * a;
			cnt %= mod;
		}
		if(i != 0)
		{
			cnt += modcomb(n - 1, i - 1, mod) * b;
			cnt %= mod;
		}
		y += (cnt * cnt) % mod;
		y %= mod;
	}
	cout << y << endl;
	return 0;
}
0