結果

問題 No.1596 Distance Sum in 2D Plane
ユーザー boatmuscles
提出日時 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);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff #

#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;
}
0