結果

問題 No.1596 Distance Sum in 2D Plane
コンテスト
ユーザー monnu
提出日時 2021-07-09 22:08:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 203 ms / 2,000 ms
コード長 952 bytes
コンパイル時間 1,551 ms
コンパイル使用メモリ 168,932 KB
実行使用メモリ 26,948 KB
最終ジャッジ日時 2024-07-01 16:35:31
合計ジャッジ時間 5,240 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
//#include <atcoder/all>
//using namespace atcoder;
using ll=long long;
using Graph=vector<vector<int>>;
#define MAX 1000000
#define MOD 1000000007
//#define INF 1000000000
#define INF 1000000000000000000

ll fac[MAX],finv[MAX],inv[MAX];

void COMinit(){
  fac[0]=fac[1]=1;
  finv[0]=finv[1]=1;
  inv[1]=1;
  for(int i=2;i<MAX;i++){
    fac[i]=(fac[i-1]*i)%MOD;
    inv[i]=MOD-inv[MOD%i]*(MOD/i)%MOD;
    finv[i]=finv[i-1]*inv[i]%MOD;
  }
}

ll COM(int n,int k){
  if(n<k){
    return 0;
  }
  if(n<0||k<0){
    return 0;
  }
  return fac[n]*(finv[k]*finv[n-k]%MOD)%MOD;
}


int main(){
  int N,M;
  cin>>N>>M;
  COMinit();
  ll ans=2*(ll)N*COM(2*N,N)%MOD;
  //cout<<ans<<endl;

  for(int i=0;i<M;i++){
    int t,x,y;
    cin>>t>>x>>y;
    if(t==1){
      ans+=MOD-COM(x+y,x)*COM(2*N-x-y-1,N-y)%MOD;
    }else{
      ans+=MOD-COM(x+y,x)*COM(2*N-x-y-1,N-x)%MOD;
    }
    ans%=MOD;
  }
  cout<<ans<<endl;
}
0