結果

問題 No.503 配列コレクション
ユーザー はまやんはまやん
提出日時 2017-05-06 12:58:42
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 985 ms / 2,000 ms
コード長 1,594 bytes
コンパイル時間 1,701 ms
コンパイル使用メモリ 167,772 KB
実行使用メモリ 11,264 KB
最終ジャッジ日時 2024-09-14 13:51:38
合計ジャッジ時間 16,016 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a;i<b;i++)





int mod = 1000000007;
int add(int x, int y) { return (x += y) >= mod ? x - mod : x; }
template<class... T> int add(int x, T... y) { return add(x, add(y...)); }
int mul(int x, int y) { return 1LL * x * y % mod; }
template<class... T> int mul(int x, T... y) { return mul(x, mul(y...)); }
int sub(int x, int y) { return add(x, mod - y); }
int modpow(int a, long long b) { int ret = 1; while (b > 0) { if (b & 1) ret = 1LL * ret * a % mod; a = 1LL * a * a % mod; b >>= 1; } return ret; }
int modinv(int a) { return modpow(a, mod - 2); }

#define FAC_MAX 1001010
int fac[FAC_MAX], ifac[FAC_MAX];
void initfac() {
    fac[0] = ifac[0] = 1;
    rep(i, 1, FAC_MAX) fac[i] = 1LL * i * fac[i - 1] % mod;
    rep(i, 1, FAC_MAX) ifac[i] = modinv(fac[i]);
}
int aCb(int a, int b) {
    if (b < 0 || a < b) return 0;
    return (1LL * fac[a] * ifac[a - b] % mod) * ifac[b] % mod;
}
int nHk(int n, int k) {
    if (n == 0 && k == 0) return 1;
    if (n <= 0 || k < 0) return 0;
    return aCb(n + k - 1, k);
}
//-----------------------------------------------------------------
int N, K, D;
//-----------------------------------------------------------------
int main() {
    cin >> N >> K >> D;

    int t = (N - 1) / (K - 1);
    int r = N - t * (K - 1);

    if (D == 1) {
        printf("%d\n", r);
        return 0;
    }

    int ans = 0;
    initfac();
    rep(k, 0, t + 1) {
        int d = mul(modpow(D, k), nHk(r - 1, t - k));
        ans = add(ans, d);
    }
    ans = mul(ans, r);
    cout << ans << endl;
}
0