結果
問題 | No.823 Many Shifts Easy |
ユーザー |
![]() |
提出日時 | 2019-06-25 22:36:05 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 34 ms / 2,000 ms |
コード長 | 1,745 bytes |
コンパイル時間 | 1,606 ms |
コンパイル使用メモリ | 167,072 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-06-23 10:27:16 |
合計ジャッジ時間 | 2,325 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 10 |
ソースコード
#include<bits/stdc++.h> using namespace std; using ll = long long; template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; } template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; } #define FOR(i,a,b) for(ll i=(a);i<(b);++i) #define ALL(v) (v).begin(), (v).end() #define p(s) cout<<(s)<<endl #define p2(s, t) cout << (s) << " " << (t) << endl #define br() p("") #define pn(s) cout << (#s) << " " << (s) << endl #define p_yes() p("Yes") #define p_no() p("No") const ll mod = 1e9 + 7; const ll inf = 1e18; const int N_MAX = 100010; ll Per[N_MAX] = {}; // n! ll Per_inv[N_MAX] = {}; //(n!)^-1 ll nCr(ll n, ll r){ if(n<r) return 0; if (n == r || r == 0) return 1; else return Per[n] * Per_inv[n-r] % mod * Per_inv[r] % mod; } // a^b mod p ll mod_pow(ll a, ll b){ if(b==0) return 1; // 肩が奇数 if(b%2==1){ return a * mod_pow(a, b-1) % mod; } else{ return mod_pow(a*a % mod, b/2) % mod; } } void prepare_nCr(){ // nCr高速化準備 Per[1] = 1; FOR(i, 2, N_MAX){ Per[i] = i * Per[i-1] % mod; } Per_inv[1] = 1; FOR(i, 2, N_MAX){ Per_inv[i] = mod_pow(Per[i], 1000000005); } } ll nPk(ll N, ll K){ ll ans = 1; FOR(i, 0, K){ ans *= N; ans %= mod; N--; } return ans; } int main(){ cin.tie(0); ios::sync_with_stdio(false); // input ll N, K; cin >> N >> K; prepare_nCr(); ll left = N*(N+1)/2 % mod * nPk(N-1, K) % mod; ll right = N*(N-1)/2 % mod * nCr(N-2, K-2) % mod * Per[K] % mod * mod_pow(2, 1000000005) % mod; ll ans = left + right; ans %= mod; p(ans); return 0; }