結果

問題 No.3579 区間積逆像
コンテスト
ユーザー tobisatis
提出日時 2026-07-03 22:03:06
言語 C#
(.NET 10.0.201)
コンパイル:
dotnet_c
実行:
/usr/bin/dotnet_wrap
結果
AC  
実行時間 144 ms / 2,000 ms
コード長 1,637 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 6,845 ms
コンパイル使用メモリ 171,344 KB
実行使用メモリ 203,992 KB
最終ジャッジ日時 2026-07-03 22:05:01
合計ジャッジ時間 10,889 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (91 ミリ秒)。
  main -> /home/judge/data/code/bin/Release/net10.0/main.dll
  main -> /home/judge/data/code/bin/Release/net10.0/publish/

ソースコード

diff #
raw source code 

#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()!.Trim().Split(' '), 0);
    return T.Parse(_input[_iter++], null);
}
#endregion

var n = I<int>();
var p = I<int>();
var c = I<int>();
var bz = new List<List<int>>();
{
    var cl = new List<int>();
    for (var i = 0; i < n; i++)
    {
        var a = I<int>();
        if (a == 0)
        {
            if (cl.Count > 0)
            {
                bz.Add(cl);
                cl = [];
            }
        }
        else cl.Add(a);
    }
    if (cl.Count > 0) bz.Add(cl);
}
int invC;
{
    var cv = 1L;
    long ck = c;
    var cp = p - 2;
    while (cp > 0)
    {
        if ((cp & 1) != 0) cv = cv * ck % p;
        ck = ck * ck % p;
        cp >>= 1;
    }
    invC = (int)cv;
}
var ans = 0L;
if (c == 0)
{
    ans = (long)n * (n - 1) / 2 + n;
    foreach (var b in bz)
    {
        var len = b.Count;
        ans -= (long)len * (len - 1) / 2 + len;
    }
}
else
{
    foreach (var b in bz)
    {
        var data = new HashMap<int, int>(){ [1] = 1 };
        var last = 1L;
        foreach (var v in b)
        {
            var next = last * v % p;
            var k = next * invC % p;
            ans += data[(int)k];
            last = next;
            data[(int)last]++;
        }
    }
}
Console.WriteLine(ans);

class HashMap<K, V> : Dictionary<K, V> where K : notnull where V : struct
{
    public new V this[K k]
    {
        get => TryGetValue(k, out var value) ? value : base[k] = default;
        set { base[k] = value; }
    }
}
0