結果

問題 No.3328 岩井ツリーグラフ
コンテスト
ユーザー tobisatis
提出日時 2025-11-01 16:41:51
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 111 ms / 2,000 ms
コード長 1,127 bytes
コンパイル時間 7,659 ms
コンパイル使用メモリ 170,152 KB
実行使用メモリ 223,044 KB
最終ジャッジ日時 2025-11-01 16:42:05
合計ジャッジ時間 11,327 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (100 ミリ秒)。
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff # 

#nullable enable

#region
var (_input, _iter) = (Array.Empty<string>(), 0);
T I<T>() where T : IParsable<T>
{
    while (_iter >= _input.Length) (_input, _iter) = (Console.ReadLine()!.Split(' '), 0);
    return T.Parse(_input[_iter++], null);
}
#endregion

const int Mod = 998244353;

long Power(long v, long p)
{
    var (res, k) = (1L, v);
    while (p > 0)
    {
        if ((p & 1) > 0) res = res * k % Mod;
        k = k * k % Mod;
        p >>= 1;
    }
    return res;
}
var inv2 = Power(2, Mod - 2);
var inv6 = Power(6, Mod - 2);

static T[] Range<T>(int n, Func<T> F) => Enumerable.Range(0, n).Select(_ => F()).ToArray();

var n = I<int>();
var yz = Range(n, I<long>);

var o = 0L;
var ty = 0L;
foreach (var y in yz)
{
    o = (o + y * (y + 1) % Mod * inv2) % Mod;
    ty += y;
}
ty %= Mod;
var ans = o * (ty + 1) % Mod;
foreach (var y in yz)
{
    var yb = y * (y + 1) % Mod * inv2 % Mod;
    var oe = (ty - y) % Mod;
    var t = y * (1 + y) % Mod * (y - 1) % Mod * inv6 % Mod;
    var yns = (oe * yb - t) % Mod;
    ans = (ans + yns) % Mod;
}
ans = ans * inv2 % Mod;
if (ans < 0) ans += Mod;
Console.WriteLine(ans);
0