結果
問題 |
No.823 Many Shifts Easy
|
ユーザー |
|
提出日時 | 2020-09-05 20:47:00 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 29 ms / 2,000 ms |
コード長 | 966 bytes |
コンパイル時間 | 2,215 ms |
コンパイル使用メモリ | 192,828 KB |
最終ジャッジ日時 | 2025-01-14 07:17:31 |
ジャッジサーバーID (参考情報) |
judge3 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 10 |
ソースコード
#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 = 1LL * k * (k-1) / 2 % MOD; cnt += tmp * perm(n-2, k-2) % MOD; } ret += i * cnt % MOD; ret %= MOD; } cout << ret << endl; return 0; }