結果

問題 No.1596 Distance Sum in 2D Plane
ユーザー boatmusclesboatmuscles
提出日時 2021-07-09 22:57:23
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 81 ms / 2,000 ms
コード長 1,596 bytes
コンパイル時間 342 ms
コンパイル使用メモリ 49,880 KB
実行使用メモリ 8,936 KB
最終ジャッジ日時 2023-09-14 10:29:44
合計ジャッジ時間 2,573 ms
ジャッジサーバーID
(参考情報)
judge13 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 9 ms
8,848 KB
testcase_01 AC 9 ms
8,860 KB
testcase_02 AC 79 ms
8,856 KB
testcase_03 AC 76 ms
8,844 KB
testcase_04 AC 78 ms
8,936 KB
testcase_05 AC 81 ms
8,844 KB
testcase_06 AC 77 ms
8,880 KB
testcase_07 AC 79 ms
8,912 KB
testcase_08 AC 77 ms
8,936 KB
testcase_09 AC 75 ms
8,856 KB
testcase_10 AC 76 ms
8,824 KB
testcase_11 AC 58 ms
8,900 KB
testcase_12 AC 57 ms
8,856 KB
testcase_13 AC 58 ms
8,852 KB
testcase_14 AC 2 ms
4,376 KB
testcase_15 AC 1 ms
4,376 KB
testcase_16 AC 1 ms
4,380 KB
testcase_17 AC 2 ms
4,380 KB
testcase_18 AC 1 ms
4,376 KB
testcase_19 AC 1 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

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