結果

問題 No.3376 Rectangle in Circle
コンテスト
ユーザー kakel-san
提出日時 2025-11-22 00:18:40
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 381 ms / 2,000 ms
コード長 2,465 bytes
コンパイル時間 8,480 ms
コンパイル使用メモリ 168,796 KB
実行使用メモリ 218,632 KB
最終ジャッジ日時 2025-11-22 00:18:57
合計ジャッジ時間 14,727 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (109 ミリ秒)。
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff # 

using System;
using static System.Console;
using System.Linq;
using System.Collections.Generic;
using System.Globalization;
using System.Runtime.CompilerServices;
using System.Runtime.Intrinsics.X86;
using System.Runtime.Intrinsics.Arm;

class Program
{
    static int NN => int.Parse(ReadLine());
    static int[] NList => ReadLine().Split().Select(int.Parse).ToArray();
    public static void Main()
    {
        Solve();
    }
    static void Solve()
    {
        var t = NN;
        var ans = new long[t];
        for (var u = 0; u < t; ++u)
        {
            var c = NList;
            var a = NList;
            ans[u] = Rect(c[0], c[1], a);
        }
        WriteLine(string.Join("\n", ans));
    }
    static long Rect(int n, int l, int[] a)
    {
        if (l % 2 == 1) return Gacha(n);
        var set = new HashSet<int>();
        foreach (var ai in a) if (ai >= l / 2) set.Add(ai);
        var count = 0;
        foreach (var ai in a) if (ai < l / 2 && set.Contains(ai + l / 2)) ++count;
        if (count < 2) return Gacha(n);
        return Gacha2(n, count);
    }
    static int mod = 998_244_353;
    static long Gacha(int n)
    {
        var dp = new long[n + 1];
        for (var i = n - 1; i >= 0; --i)
        {
            dp[i] = (dp[i + 1] + n * Exp(n - i, mod - 2, mod) % mod) % mod;
        }
        return dp[0];
    }
    static long Gacha2(int n, int p)
    {
        var dp1 = new long[p + 1];
        dp1[p - 1] = n * Exp(p - 1, mod - 2, mod);
        for (var i = p - 2; i >= 0; --i)
        {
            var rev2ni = Exp(2 * p - i - 2, mod - 2, mod);
            dp1[i] = (2 * (p - i - 1) * dp1[i + 1] + n) % mod * rev2ni % mod;
        }
        var dp0 = new long[p + 1];
        dp0[p] = (dp1[p - 1] + n * Exp(p, mod - 2, mod)) % mod;
        for (var i = p - 1; i >= 0; --i)
        {
            var rev2ni = Exp(2 * p - i, mod - 2, mod);
            dp0[i] = (i * (i == 0 ? 0 : dp1[i - 1]) + 2 * (p - i) * dp0[i + 1] + n) % mod * rev2ni % mod;
        }
        return dp0[0];
    }
    static long Exp(long n, long p, int mod)
    {
        long _n = n % mod;
        var _p = p;
        var result = 1L;
        if ((_p & 1) == 1) result *= _n;
        while (_p > 0)
        {
            _n = _n * _n % mod;
            _p >>= 1;
            if ((_p & 1) == 1) result = result * _n % mod;
        }
        return result;
    }
}
0