結果

問題 No.2651 [Cherry 6th Tune B] $\mathbb{C}$omplex комбинат
ユーザー tassei903
提出日時 2024-02-23 22:10:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,124 ms / 2,500 ms
コード長 1,346 bytes
コンパイル時間 227 ms
コンパイル使用メモリ 82,424 KB
実行使用メモリ 127,892 KB
最終ジャッジ日時 2024-11-19 08:30:19
合計ジャッジ時間 36,725 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes")
no = lambda :print("no");No = lambda :print("No")
#######################################################################
mod = 998244353
def mul(a, b):
    return ((a[0] * b[0] - a[1] * b[1]) % mod, (a[0] * b[1] + a[1] * b[0]) % mod)
def add(a, b):
    return ((a[0] + b[0]) % mod, (a[1] + b[1]) % mod)

def norm(a):
    return (a[0] * a[0] + a[1] * a[1]) % mod

def div(a, b):
    return mul(mul(a, (b[0], -b[1])) , (pow(norm(b), mod - 2, mod), 0))

for i in range(ni()):
    n = ni()
    p = [na() for _ in range(n)]
    normP = [norm(x) for x in p]
    B = 0
    for i in range(n):
        B = (B + normP[i]) % mod
    C = 0
    for i in range(n):
        C = (C + pow(normP[i], mod - 2, mod)) % mod
    D = (2 * B * C) % mod
    D -= n * 2
    D *= pow(2, mod - 2, mod)
    for i in range(n):
        p[i] = mul(p[i], p[i])
        p[i] = div(p[i], (normP[i], 0))
    sum = (0, 0)
    inv_p = [div((1, 0), x) for x in p]
    sumP = (0, 0)
    for i in range(n):
        sumP = add(sumP, p[i])
    sumP_inv = (0, 0)
    for i in range(n):
        sumP_inv = add(sumP_inv, inv_p[i])
    ans = mul(sumP, sumP_inv)
    ans = add(ans, (-n, 0))
    print((D-ans[0])%mod)
0