結果

問題 No.1596 Distance Sum in 2D Plane
コンテスト
ユーザー boatmuscles
提出日時 2021-07-10 17:34:41
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 88 ms / 2,000 ms
コード長 950 bytes
コンパイル時間 587 ms
コンパイル使用メモリ 51,200 KB
最終ジャッジ日時 2025-01-23 00:26:05
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:10:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   10 |     scanf("%u %u", &N, &M);
      |     ~~~~~^~~~~~~~~~~~~~~~~
main.cpp:24:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   24 |         scanf("%u %u %u", &t, &x, &y);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include<cstdio>
#include<vector>
#include<atcoder/modint>
using namespace std;
using namespace atcoder;
using Mint = modint1000000007;

int main(){
    unsigned N, M;
    scanf("%u %u", &N, &M);
    vector<Mint> kaijou((N<<1)+1);
    kaijou[0] = Mint::raw(1);
    for (unsigned i = 0; i < N<<1; i++){
        kaijou[i+1] = kaijou[i]*Mint::raw(i+1);
    }
    vector<Mint> rev_kaijou((N<<1)+1);
    rev_kaijou[N<<1] = kaijou[N<<1].inv();
    for (unsigned i = N<<1; i > 0; i--){
        rev_kaijou[i-1] = rev_kaijou[i]*Mint::raw(i);
    }
    Mint answer = Mint::raw(N<<1)*kaijou[N<<1]*rev_kaijou[N]*rev_kaijou[N];
    for (unsigned i = 0; i < M; i++){
        unsigned t, x, y;
        scanf("%u %u %u", &t, &x, &y);
        answer -= kaijou[x+y]*rev_kaijou[x]*rev_kaijou[y]*kaijou[(N<<1) - x - y - 1]*(t == 1 ? rev_kaijou[N - x - 1]*rev_kaijou[N - y] :(rev_kaijou[N - y - 1]*rev_kaijou[N - x]));
    }
    printf("%u\n", answer.val());
	return 0;
}
0