結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 132 ms
6,276 KB
testcase_02 AC 92 ms
5,228 KB
testcase_03 AC 68 ms
4,692 KB
testcase_04 AC 18 ms
4,380 KB
testcase_05 AC 62 ms
4,376 KB
testcase_06 AC 131 ms
6,188 KB
testcase_07 AC 39 ms
4,380 KB
testcase_08 AC 38 ms
4,380 KB
testcase_09 AC 108 ms
5,744 KB
testcase_10 AC 110 ms
5,608 KB
testcase_11 AC 93 ms
5,132 KB
testcase_12 AC 86 ms
5,236 KB
testcase_13 AC 80 ms
4,856 KB
testcase_14 AC 65 ms
4,712 KB
testcase_15 AC 35 ms
4,376 KB
testcase_16 AC 72 ms
4,548 KB
testcase_17 AC 36 ms
4,376 KB
testcase_18 AC 78 ms
4,856 KB
testcase_19 AC 94 ms
5,264 KB
testcase_20 AC 97 ms
5,480 KB
testcase_21 AC 99 ms
5,472 KB
testcase_22 AC 35 ms
4,376 KB
testcase_23 AC 110 ms
5,740 KB
testcase_24 AC 69 ms
4,548 KB
testcase_25 AC 73 ms
4,640 KB
testcase_26 AC 80 ms
4,856 KB
testcase_27 AC 94 ms
5,272 KB
testcase_28 AC 91 ms
5,268 KB
testcase_29 AC 33 ms
4,380 KB
testcase_30 AC 24 ms
4,376 KB
testcase_31 AC 98 ms
5,532 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