結果

問題 No.823 Many Shifts Easy
ユーザー Y17Y17
提出日時 2019-04-26 23:16:11
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 45 ms / 2,000 ms
コード長 866 bytes
コンパイル時間 1,177 ms
コンパイル使用メモリ 145,216 KB
実行使用メモリ 4,384 KB
最終ジャッジ日時 2023-08-16 17:22:26
合計ジャッジ時間 2,076 ms
ジャッジサーバーID
(参考情報)
judge13 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
4,376 KB
testcase_01 AC 3 ms
4,384 KB
testcase_02 AC 3 ms
4,384 KB
testcase_03 AC 44 ms
4,384 KB
testcase_04 AC 2 ms
4,384 KB
testcase_05 AC 29 ms
4,380 KB
testcase_06 AC 45 ms
4,384 KB
testcase_07 AC 3 ms
4,380 KB
testcase_08 AC 43 ms
4,380 KB
testcase_09 AC 8 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

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

#define int long long

#define MOD 1000000007

int kai[100010];

void init(){
    kai[0] = 1;
    for(int i = 1;i < 100010;i++){
        kai[i] = kai[i-1] * i % MOD;
    }
}

int pow_mod(int n, int m){
    int ans = 1;
    while(m > 0){
        if(m & 1) ans = ans * n % MOD;
        n = n * n % MOD;
        m >>= 1;
    }
    return ans;
}

int nPr(int n, int r){
    return kai[n] * pow_mod(kai[n-r], MOD-2) % MOD;
}

int souwa(int n){
    return (n*(n+1)) % MOD * pow_mod(2, MOD-2) % MOD;
}

signed main(){
    int n, k;
    cin >> n >> k;
    init();

    int ans = 0;
    for(int i = 1;i < n;i++){
        int cnt = (nPr(n-1, k) + souwa(k-1)*nPr(n-2, k-2) % MOD) % MOD;
        ans = ((cnt * i) % MOD + ans) % MOD;
    }

    ans = (ans + (nPr(n-1, k) * n) % MOD) % MOD;
    cout << ans << endl;
    return 0;
}

0