結果

問題 No.503 配列コレクション
ユーザー gigimegigime
提出日時 2017-04-07 23:47:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 15 ms / 2,000 ms
コード長 1,132 bytes
コンパイル時間 1,646 ms
コンパイル使用メモリ 168,208 KB
実行使用メモリ 17,016 KB
最終ジャッジ日時 2024-07-16 03:20:43
合計ジャッジ時間 2,406 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

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

#define FOR(i,l,r) for(int i = (int) (l);i < (int) (r);i++)
#define ALL(x) x.begin(),x.end()
template<typename T> bool chmax(T& a,const T& b){ return a < b ? (a = b,true) : false; }
template<typename T> bool chmin(T& a,const T& b){ return b < a ? (a = b,true) : false; }
typedef long long ll;

ll N,K,D;
const ll MAX = 1700000;
ll fact [MAX];
const ll MOD = 1e9 + 7;

ll mod_pow(ll x,ll y = MOD - 2)
{
	ll res = 1;
	while(y){
		if(y & 1) (res *= x) %= MOD;
		(x *= x) %= MOD;
		y >>= 1;
	}
	return res;
}

int main()
{
	scanf("%lld%lld%lld",&N,&K,&D);
	ll a = N % (K - 1);
	if(a == 0) a += K - 1;
	ll t = (N - a) / (K - 1);
	if(a == 1){
		printf("%lld\n",mod_pow(D,t));
		return 0;
	}
	if(D == 1){
		printf("%lld\n",a);
		return 0;
	}

	fact [0] = 1;
	for(ll i = 1;i < MAX;i++){
		fact [i] = fact [i - 1] * i % MOD;
	}

	ll ans = 0;
	for(ll i = 0;i <= t;i++){
		ll r = fact [(a - 2) + (t - i)];
		(r *= mod_pow(fact [t - i])) %= MOD;
		(r *= mod_pow(fact [a - 2])) %= MOD;
		(r *= mod_pow(D,i)) %= MOD;
		(ans += r) %= MOD;
	}
	(ans *= a) %= MOD;

	printf("%lld\n",ans);

	return 0;
}
0