結果
問題 | No.823 Many Shifts Easy |
ユーザー |
|
提出日時 | 2020-09-05 20:40:27 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 960 bytes |
コンパイル時間 | 2,106 ms |
コンパイル使用メモリ | 193,112 KB |
最終ジャッジ日時 | 2025-01-14 07:17:22 |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 7 WA * 3 |
ソースコード
#include <bits/stdc++.h>using namespace std;#define REP(i,n) for(int i=0; i<(int)(n); i++)const long long MOD = 1e9 + 7;long long f[100000+1];long long g[100000+1];long long modpow(long long x, long long p, long long mod) {long long ret = 1;while (p) {if (p & 1)ret = ret * x % mod;x = x * x % mod;p >>= 1;}return ret;}long long perm(long long n, long long k) {return n >= k ? f[n] * g[n-k] % MOD : 0;}int main() {ios_base::sync_with_stdio(0);cin.tie(0);int n, k;cin >> n >> k;f[0] = g[0] = 1;for (int i = 1; i <= 100000; i++) {f[i] = i * f[i-1] % MOD;g[i] = modpow(f[i], MOD-2, MOD);}long long ret = 0;for (int i = 1; i <= n; i++) {long long cnt = perm(n-1, k);if (i+1 <= n) {long long tmp = k * (k-1) / 2 % MOD;cnt += tmp * perm(n-2, k-2) % MOD;}ret += i * cnt % MOD;ret %= MOD;}cout << ret << endl;return 0;}