結果

問題 No.823 Many Shifts Easy
コンテスト
ユーザー ttttan2
提出日時 2019-04-26 22:46:06
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 56 ms / 2,000 ms
コード長 1,288 bytes
コンパイル時間 1,158 ms
コンパイル使用メモリ 159,980 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-25 05:21:42
合計ジャッジ時間 1,761 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<pii,int> ppii;
typedef pair<ll,ll> pll;
typedef tuple<ll,ll,ll> tl;
ll mod=1000000007;
ll gcd(ll a,ll b){
  if(a<b)swap(a,b);
  if(a%b==0)return b;
  return gcd(b,a%b);
}
ll lcm(ll a,ll b){
  ll ret=a*b;
  ll u=gcd(a,b);
  return ret/u;
}
ll gyaku(ll n){
  ll u=mod-2;
  ll ret=1;
  ll now=n;
  while(u>0){
    if(u%2){
      ret*=now;
      ret%=mod;
    }
    now*=now;
    now%=mod;
    u/=2;
  }
  return ret;
}
ll nck(ll n,ll k){
  ll ret=1;
  for(int i=0;i<k;i++){
    ret*=n-i;
    ret%=mod;
    ret*=gyaku(i+1);
    ret%=mod;
  }
  return ret;
}
ll npk(ll n,ll k){
  ll ret=1;
  for(int i=0;i<k;i++){
    ret*=n-i;
    ret%=mod;
  }
  return ret;
}
ll kai(ll n,ll k){
  ll ret=1;
  ll now=n;
  while(k>0){
    if(k%2){
      ret*=now;
      ret%=mod;
    }
    now*=now;
    now%=mod;
    k/=2;
  }
  return ret;
}
int main(){
  ll n,k;cin>>n>>k;
  if(n==1){
    cout<<0<<endl;
    return 0;
  }
  ll sum=0;
  if(k<n){
    sum=npk(n-1,k);
  }
  ll u=0;
  if(k>1){
    u=nck(n-2,k-2);
  for(int i=1;i<=k;i++){
    u*=i;
    u%=mod;
  }
  u*=gyaku(2);
  u%=mod;
  }
  ll ans=0;
  for(int i=1;i<n;i++)ans+=i;
  ans*=sum+u;
  ans%=mod;
  ans+=n*sum;
  cout<<ans%mod<<endl;
}
0