結果

問題 No.1684 Find Brackets
コンテスト
ユーザー TKTYI
提出日時 2021-09-17 23:18:39
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 476 ms / 2,000 ms
コード長 1,329 bytes
コンパイル時間 3,624 ms
コンパイル使用メモリ 232,148 KB
実行使用メモリ 7,516 KB
最終ジャッジ日時 2024-06-29 22:06:34
合計ジャッジ時間 7,268 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
typedef long long int ll;
typedef long double ld;
#define FOR(i,l,r) for(ll i=l;i<r;i++)
#define REP(i,n) FOR(i,0,n)
#define RFOR(i,l,r) for(ll i=r-1;i>=l;i--)
#define RREP(i,n) RFOR(i,0,n)
#define ALL(x) x.begin(),x.end()
#define P pair<ll,ll>
#define F first
#define S second
#define BS(A,x) binary_search(ALL(A),x)
#define LB(A,x) (ll)(lower_bound(ALL(A),x)-A.begin())
#define UB(A,x) (ll)(upper_bound(ALL(A),x)-A.begin())
#define COU(A,x) UB(A,x)-LB(A,x)
template<typename T>using min_priority_queue=priority_queue<T,vector<T>,greater<T>>;
using mint=modint1000000007;
mint K[1000001];
signed main(){
  ll N,M;cin>>N>>M;mint ans=0;K[0]=1;FOR(i,1,1000001)K[i]=K[i-1]*i;
  if(2*M<N){
    mint sum=0;
    REP(i,N/2+1){
      mint res=2*(N/2+1-i)*(N-i);res*=K[N]/K[i]/K[N-i]*(N-2*i+1);res/=N-i+1;sum+=2*res;
    }
    if(N%2==0)REP(i,N/2+1){
      mint res=2*(N-i);res*=K[N]/K[i]/K[N-i]*(N-2*i+1);res/=N-i+1;sum-=res;
    }
    REP(i,M){
      mint res=2*(M-i)*(N-i);
      res*=K[N]/K[i]/K[N-i]*(N-2*i+1);res/=N-i+1;
      ans+=res;
    }
    ans=sum-ans;
  }
  else{
    REP(i,N-M+1){
      mint res=2*(N-M+1-i)*(N-i);
      res*=K[N]/K[i]/K[N-i]*(N-2*i+1);res/=N-i+1;
      ans+=res;
    }
  }
  cout<<ans.val()<<endl;
  return 0;
}
0