結果
問題 | No.823 Many Shifts Easy |
ユーザー |
![]() |
提出日時 | 2019-04-26 21:40:19 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 4 ms / 2,000 ms |
コード長 | 1,667 bytes |
コンパイル時間 | 1,433 ms |
コンパイル使用メモリ | 168,808 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-11-25 02:28:16 |
合計ジャッジ時間 | 1,915 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 10 |
ソースコード
#include <bits/stdc++.h>using namespace std;typedef long long ll;typedef long double dd;#define i_7 (ll)(1E9+7)//#define i_7 998244353#define i_5 i_7-2ll mod(ll a){ll c=a%i_7;if(c>=0)return c;return c+i_7;}typedef pair<ll,ll> l_l;ll inf=(ll)1E17;#define rep(i,l,r) for(ll i=l;i<=r;i++)#define pb push_backll max(ll a,ll b){if(a<b)return b;else return a;}ll min(ll a,ll b){if(a>b)return b;else return a;}void Max(ll &pos,ll val){pos=max(pos,val);}//Max(dp[n],dp[n-1]);void Min(ll &pos,ll val){pos=min(pos,val);}void Add(ll &pos,ll val){pos=mod(pos+val);}dd EPS=1E-9;#define fastio ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);///////////////////////////////////////#define N 100004ll kai[N];ll kai2[N];ll po(ll i,ll p){if(p==0)return 1;else if(p==1)return mod(i);else if(p%2==0)return po(mod(mod(i)*mod(i)),p/2);else return mod(mod(i)*po(i,p-1));}void calc(){kai[0]=1;kai2[0]=1;rep(i,1,N-1){kai[i]=mod(kai[i-1]*i);}kai2[N-1]=po(kai[N-1],i_5);for(ll i=N-2;i>=0;i--){kai2[i]=mod(kai2[i+1]*(i+1));}}ll comb(ll n,ll k){if(n<k)return 0;else if(n==0)return 1;else return mod(mod(kai[n]*kai2[n-k])*kai2[k]);}ll bunbo(ll n){return po(n,i_5);}int main(){fastiocalc();ll n,k;cin>>n>>k;ll ans=0;ans=mod((n-k)*bunbo(n));ans=mod(ans*mod(n*(n+1)/2));ll ans2=0;ans2=mod(mod(k*(k-1))*mod(bunbo(n)*bunbo(n-1)));ans2=mod(ans2*bunbo(2));ans2=mod(ans2*mod((n-1)*n/2));ans=mod(ans+ans2);ans=mod(ans*mod(comb(n,k)*kai[k]));cout<<ans<<endl;return 0;}