結果
問題 | No.1596 Distance Sum in 2D Plane |
ユーザー |
|
提出日時 | 2021-07-09 22:57:23 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 129 ms / 2,000 ms |
コード長 | 1,596 bytes |
コンパイル時間 | 780 ms |
コンパイル使用メモリ | 48,384 KB |
最終ジャッジ日時 | 2025-01-22 22:27:16 |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
コンパイルメッセージ
main.cpp: In function ‘int main()’: main.cpp:23:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 23 | scanf("%llu %u", &N, &M); | ~~~~~^~~~~~~~~~~~~~~~~~~ main.cpp:40:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 40 | scanf("%u %u %u", &t, &x, &y); | ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
ソースコード
#include<cstdio> #include<vector> using namespace std; using ll = long long; using ull = unsigned long long; const static int MOD = 1000000007; ull RepeatSquaring(ull N, ull P, ull mod){ if(P == 0 || N == 1) return 1; if(N == 0) return 0; if(N < 0) return P & 1 ? -RepeatSquaring(-N, P, mod) : RepeatSquaring(-N, P, mod); if(!(P & 1)){ ll t = RepeatSquaring(N, P >> 1, mod); return t*t %mod; } //Pが偶数のときN^(P/2)を求めてからそれの2乗を返す return N*RepeatSquaring(N, P - 1, mod) % mod; //Pが奇数のときN^((P - 1)/2)を求めてからそれの2乗を返す } int main(){ ull N; unsigned M; scanf("%llu %u", &N, &M); vector<ull> kaijou((N<<1)+1); kaijou[0] = 1; for (ull i = 0; i < N<<1; i++) { kaijou[i+1] = kaijou[i]*(i+1)%MOD; } vector<ull> rev_kaijou((N<<1)+1); rev_kaijou[N<<1] = RepeatSquaring(kaijou[N<<1], MOD - 2, MOD); for (ll i = (N<<1)-1; i >= 0; i--) { rev_kaijou[i] = rev_kaijou[i+1]*(i+1)%MOD; } ull answer = (N<<1)*kaijou[N<<1]%MOD*rev_kaijou[N]%MOD*rev_kaijou[N]%MOD; for (unsigned i = 0; i < M; i++) { unsigned t, x, y; scanf("%u %u %u", &t, &x, &y); answer = answer + MOD - (t == 1 ? kaijou[x+y]*rev_kaijou[x]%MOD*rev_kaijou[y]%MOD*kaijou[(N<<1) - x - y - 1]%MOD*rev_kaijou[N - x - 1]%MOD*rev_kaijou[N - y]%MOD : kaijou[x+y]*rev_kaijou[x]%MOD*rev_kaijou[y]%MOD*kaijou[(N<<1) - x - y - 1]%MOD*rev_kaijou[N - y - 1]%MOD*rev_kaijou[N - x]%MOD); if(answer >= MOD) answer -= MOD; } printf("%llu\n", answer); return 0; }