結果

問題 No.1596 Distance Sum in 2D Plane
ユーザー kenta255
提出日時 2021-07-10 01:26:21
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 254 ms / 2,000 ms
コード長 1,146 bytes
コンパイル時間 2,572 ms
コンパイル使用メモリ 197,276 KB
最終ジャッジ日時 2025-01-22 23:27:55
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
//#include <atcoder/modint>
using namespace std;
//using namespace atcoder;
//using mint = modint1000000007;
typedef long long ll;
#define FOR(I,A,B) for(ll I = ll(A); I < ll(B); ++I)

class comb{
	vector<ll> f,fr;
	ll MOD_;
	public:
	//a^(p-1) = 1 (mod p)(p->Prime numbers)
	//a^(p-2) = a^(-1)
	ll calc(ll a,ll b,ll p){//a^(b) mod p   
		if(b==0)return 1;
		ll y = calc(a,b/2,p);y=(y*y)%p;
		if(b & 1) y = (y * a) % p;
		return y;
	}
	void init(ll n,ll mod){//input max_n
		MOD_ = mod;
		f.resize(n+1);
		fr.resize(n+1);
		f[0]=fr[0]=1;
		for(ll i=1;i<n+1;i++){
			f[i] = (f[i-1] * i) % mod;
		}
		fr[n] = calc(f[n],mod-2,mod);
		for(ll i=n-1;i>=0;i--){
			fr[i] = fr[i+1] * (i+1) % mod;
		}
	}
	ll nCr(ll n,ll r){
		if(n<0||r<0||n<r)return 0;
		return f[n] * fr[r] % MOD_ * fr[n-r] % MOD_;
	}//nHr = n+r-1Cr
};

int main(){
	ll N,M;
	ll MOD = 1000000007;
	cin >> N >> M;
	comb co;
	co.init(2*N+8,MOD);
	ll ans = 2LL * N * co.nCr(2*N,N) % MOD;

	FOR(i,0,M){
		ll t,x,y;
		cin >> t >> x >> y;
		ll sub = co.nCr(x+y,x) * co.nCr((N-x)+(N-y)-1,(N-x)-(t==1)) % MOD;
		( ans += MOD - sub ) %= MOD;
	}
	cout << ans << endl;
}

0