#959181 (C++23) No.2651 [Cherry 6th Tune B] $\mathbb{C}$omplex комбинат

• 問題: No.2651 [Cherry 6th Tune B] $\mathbb{C}$omplex комбинат
• ユーザー: hitonanode
• 提出日時: 2024-03-10 18:22:54
• 言語: C++23 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 486 ms / 2,500 ms
• コード長: 1,637 bytes
• コンパイル時間: 1,106 ms
• コンパイル使用メモリ: 85,704 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-09-29 21:44:09
• 合計ジャッジ時間: 17,272 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	434 ms; 5,248 KB
testcase_03	AC	422 ms; 5,248 KB
testcase_04	AC	420 ms; 5,248 KB
testcase_05	AC	426 ms; 5,248 KB
testcase_06	AC	426 ms; 5,248 KB
testcase_07	AC	426 ms; 5,248 KB
testcase_08	AC	427 ms; 5,248 KB
testcase_09	AC	426 ms; 5,248 KB
testcase_10	AC	423 ms; 5,248 KB
testcase_11	AC	420 ms; 5,248 KB
testcase_12	AC	426 ms; 5,248 KB
testcase_13	AC	414 ms; 5,248 KB
testcase_14	AC	422 ms; 5,248 KB
testcase_15	AC	430 ms; 5,248 KB
testcase_16	AC	419 ms; 5,248 KB
testcase_17	AC	414 ms; 5,248 KB
testcase_18	AC	423 ms; 5,248 KB
testcase_19	AC	419 ms; 5,248 KB
testcase_20	AC	486 ms; 5,248 KB
testcase_21	AC	127 ms; 5,248 KB
testcase_22	AC	223 ms; 5,248 KB
testcase_23	AC	288 ms; 5,248 KB
testcase_24	AC	346 ms; 5,248 KB
testcase_25	AC	24 ms; 5,248 KB
testcase_26	AC	203 ms; 5,248 KB
testcase_27	AC	416 ms; 5,248 KB
testcase_28	AC	310 ms; 5,248 KB
testcase_29	AC	132 ms; 5,248 KB
testcase_30	AC	30 ms; 5,248 KB
testcase_31	AC	346 ms; 5,248 KB
testcase_32	AC	53 ms; 5,248 KB
testcase_33	AC	321 ms; 5,248 KB
testcase_34	AC	191 ms; 5,248 KB
testcase_35	AC	371 ms; 5,248 KB
testcase_36	AC	418 ms; 5,248 KB
testcase_37	AC	417 ms; 5,248 KB
testcase_38	AC	412 ms; 5,248 KB

ソースコード

#include <iostream>
using namespace std;

#include <atcoder/modint>
using mint = atcoder::modint998244353;

struct C {
    mint real;
    mint imag;

    C() : real(0), imag(0) {}

    C(mint real, mint imag) : real(real), imag(imag) {}

    C operator+(const C &rhs) const {
        return C(real + rhs.real, imag + rhs.imag);
    }

    C operator-(const C &rhs) const {
        return C(real - rhs.real, imag - rhs.imag);
    }

    C operator*(const C &rhs) const {
        return C(real * rhs.real - imag * rhs.imag, real * rhs.imag + imag * rhs.real);
    }

    C operator/(const C &rhs) const {
        mint norm = rhs.real * rhs.real + rhs.imag * rhs.imag;
        mint real_ = real * rhs.real + imag * rhs.imag;
        mint imag_ = imag * rhs.real - real * rhs.imag;
        return C(real_ / norm, imag_ / norm);
    }

    C conj() const {
        return C(real, -imag);
    }
};

// i != j |zi|^2 / |zj|^2 - (zi / zi*) (zj* / zj)

int main() {

    int T;
    cin >> T;
    while (T--) {
        int N;
        cin >> N;

        C sum_of_l2;
        C sum_of_inv_of_l2;
        C sum_of_z_div_z_conj;
        C sum_of_z_conj_div_z;
        while (N--) {
            int x, y;
            cin >> x >> y;
            C z(x, y);
            sum_of_l2 = sum_of_l2 + z * z.conj();
            sum_of_inv_of_l2 = sum_of_inv_of_l2 + C(1, 0) / (z * z.conj());

            sum_of_z_div_z_conj = sum_of_z_div_z_conj + z / z.conj();
            sum_of_z_conj_div_z = sum_of_z_conj_div_z + z.conj() / z;
        }

        cout << (sum_of_l2 * sum_of_inv_of_l2 - sum_of_z_div_z_conj * sum_of_z_conj_div_z).real.val() << '\n';
    }
}