結果

問題 No.3376 Rectangle in Circle
コンテスト
ユーザー 👑 loop0919
提出日時 2025-10-18 20:00:53
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,816 bytes
コンパイル時間 2,945 ms
コンパイル使用メモリ 291,720 KB
実行使用メモリ 813,824 KB
最終ジャッジ日時 2025-11-21 20:51:15
合計ジャッジ時間 9,160 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 2 WA * 6 MLE * 1 -- * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;

using mint = atcoder::modint998244353;

constexpr int MAX_N = 5000;
vector<mint> inv;

void init() {
    inv = vector<mint>(MAX_N + 1);
    for (int i = 1; i <= MAX_N; i++) {
        inv[i] = 1 / mint(i);
    }
}

mint exp_all(int n) {
    mint e = 0;
    for (int i = 1; i <= n; i++) {
        e += mint(n) * inv[i];
    }
    return e;
}

void solve() {
    int N, L;
    cin >> N >> L;

    set<int> D;
    for (int i = 0; i < N; i++) {
        int d;
        cin >> d;
        D.insert(d);
    }

    if (L % 2 == 1) {
        cout << exp_all(N).val() << "\n";
        return;
    }

    int diameters = 0, normals = 0;
    for (int d: D) {
        int opposition = (d + L / 2) % L;
        D.contains(opposition) ? diameters++ : normals++;
    }
    diameters /= 2;

    if (diameters <= 1) {
        cout << exp_all(N).val() << "\n";
        return;
    }

    vector dp(3, vector(diameters + 1, vector<mint>(normals + 1, 0)));

    for (int i = 1; i >= 0; i--) {
        for (int j = diameters; j >= 0; j--) {
            for (int k = normals; k >= 0; k--) {
                int remain = N - i - j - k;
                dp[i][j][k] = mint(i + j + k) * inv[remain];

                if (i < j) {
                    dp[i][j][k] += (dp[i + 1][j][k] + 1) * (j - i) * inv[remain];
                }
                if (j < diameters) {
                    dp[i][j][k] += (dp[i][j + 1][k] + 1) * 2 * (diameters - j) * inv[remain];
                }
                if (k < normals) {
                    dp[i][j][k] += (dp[i][j][k + 1] + 1) * (normals - k) * inv[remain];
                }
            }
        }
    }

    cout << dp[0][0][0].val() << "\n";
}

int main() {
    init();

    int T;
    cin >> T;

    while (T--) solve();
}
0