結果

問題 No.1127 変形パスカルの三角形
ユーザー Kome_soudou
提出日時 2022-11-10 02:04:37
言語 C++14
(gcc 13.3.0 + boost 1.87.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

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