結果
問題 | No.1596 Distance Sum in 2D Plane |
ユーザー |
![]() |
提出日時 | 2021-07-09 21:39:25 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 255 ms / 2,000 ms |
コード長 | 1,504 bytes |
コンパイル時間 | 1,733 ms |
コンパイル使用メモリ | 172,144 KB |
実行使用メモリ | 9,688 KB |
最終ジャッジ日時 | 2024-07-01 15:34:36 |
合計ジャッジ時間 | 5,970 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
#include <bits/stdc++.h> using namespace std; const long long MOD = 1000000007; long long modsub(long long a, long long b){ return (a % MOD - b % MOD + MOD) % MOD; } long long modpow(long long a, long long b){ long long ans = 1; while (b > 0){ if (b % 2 == 1){ ans *= a; ans %= MOD; } a *= a; a %= MOD; b /= 2; } return ans; } long long modinv(long long a){ return modpow(a, MOD - 2); } vector<long long> mf = {1}; vector<long long> mfi = {1}; long long modfact(int n){ if (mf.size() > n){ return mf[n]; } else { for (int i = mf.size(); i <= n; i++){ long long next = mf.back() * i % MOD; mf.push_back(next); mfi.push_back(modinv(next)); } return mf[n]; } } long long modfactinv(int n){ if (mfi.size() > n){ return mfi[n]; } else { return modinv(modfact(n)); } } long long modbinom(int n, int k){ if (n < 0 || k < 0 || k > n){ return 0; } else { return modfact(n) * modfactinv(k) % MOD * modfactinv(n - k) % MOD; } } int main(){ int N, M; cin >> N >> M; long long ans = modbinom(N * 2, N) * N * 2 % MOD; for (int i = 0; i < M; i++){ int t, x, y; cin >> t >> x >> y; if (t == 1){ long long a = modbinom(x + y, x); long long b = modbinom(N * 2 - x - y - 1, N - y); ans += MOD - a * b % MOD; ans %= MOD; } if (t == 2){ long long a = modbinom(x + y, x); long long b = modbinom(N * 2 - x - y - 1, N - x); ans += MOD - a * b % MOD; ans %= MOD; } } cout << ans << endl; }