結果
問題 | No.503 配列コレクション |
ユーザー |
![]() |
提出日時 | 2019-12-12 18:32:47 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 30 ms / 2,000 ms |
コード長 | 1,520 bytes |
コンパイル時間 | 943 ms |
コンパイル使用メモリ | 111,020 KB |
実行使用メモリ | 34,560 KB |
最終ジャッジ日時 | 2024-06-25 12:28:19 |
合計ジャッジ時間 | 2,070 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 25 |
ソースコード
#include <cstdio>#include <cstring>#include <iostream>#include <string>#include <cmath>#include <bitset>#include <vector>#include <map>#include <set>#include <queue>#include <deque>#include <algorithm>#include <complex>#include <unordered_map>#include <unordered_set>#include <random>#include <cassert>#include <fstream>#include <utility>#include <functional>#include <time.h>#include <stack>#include <array>#define popcount __builtin_popcountusing namespace std;typedef long long int ll;typedef pair<int, int> P;const ll MOD=1e9+7;ll powmod(ll a, ll k){ll ap=a, ans=1;while(k){if(k&1){ans*=ap;ans%=MOD;}ap=ap*ap;ap%=MOD;k>>=1;}return ans;}ll inv(ll a){return powmod(a, MOD-2);}ll f[2000020], invf[2000020];void fac(int n){f[0]=1;for(ll i=1; i<=n; i++) f[i]=f[i-1]*i%MOD;invf[n]=inv(f[n]);for(ll i=n-1; i>=0; i--) invf[i]=invf[i+1]*(i+1)%MOD;}ll comb(int x, int y){if(!(0<=y && y<=x)) return 0;return f[x]*invf[y]%MOD*invf[x-y]%MOD;}int main(){int n, k;ll d;cin>>n>>k>>d;int r=n%(k-1);if(r==0) r=k-1;if(d==1){cout<<r<<endl;return 0;}if(r==1){cout<<powmod(d, (n-r)/(k-1))<<endl;return 0;}fac(2*n);ll ans=0, p=1;for(int i=0; i<=(n-r)/(k-1); i++){(ans+=comb((n-r)/(k-1)-i+r-2, r-2)*r%MOD*p)%=MOD;(p*=d)%=MOD;}cout<<ans<<endl;return 0;}