結果

問題 No.3376 Rectangle in Circle
コンテスト
ユーザー 👑 loop0919
提出日時 2025-10-18 14:18:58
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,605 bytes
コンパイル時間 2,826 ms
コンパイル使用メモリ 292,088 KB
実行使用メモリ 152,100 KB
最終ジャッジ日時 2025-11-21 20:51:06
合計ジャッジ時間 7,312 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 3 TLE * 1 -- * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using mint = atcoder::modint998244353;

mint exp_all(int n) {
    mint e = 0;
    for (int i = 1; i <= n; i++) {
        e += mint(n) / 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--) {
                mint remain = N - i - j - k;
                dp[i][j][k] = mint(i + j + k) / remain;

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

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

int main() {
    int T;
    cin >> T;

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