結果

問題 No.3257 +|+
ユーザー tobisatis
提出日時 2025-09-05 23:23:05
言語 C#
(.NET 8.0.404)
結果
WA  
実行時間 -
コード長 1,424 bytes
コンパイル時間 11,567 ms
コンパイル使用メモリ 170,096 KB
実行使用メモリ 348,352 KB
最終ジャッジ日時 2025-09-05 23:24:09
合計ジャッジ時間 61,504 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 WA * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (113 ミリ秒)。
  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 = Array.Empty<string>();
var _iter = 0;
string String()
{
    while (_iter >= _input.Length) (_input, _iter) = (Console.ReadLine()!.Split(' '), 0);
    return _input[_iter++];
}
T I<T>() where T : IParsable<T> => T.Parse(String(), null);
#endregion

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

var n = I<int>();
var az = Range(n, I<int>);
var ans = 0L;
var bag = new Bag<(long, int)>();
for (var i = 1; i <= n; i++)
{
    var a = az[i - 1];
    var d = 0L;
    var k = 0;
    while (true)
    {
        d += i;
        k++;
        if (d > 200000) break;
        var v = d - a;
        ans += bag[(v, k)];
        bag.Add((-v, k));
    }
}
Console.WriteLine(ans);

class Bag<K> where K : notnull
{
    readonly Dictionary<K, int> _dictionary = new();
    public void Add(K key)
    {
        var dictionary = _dictionary;
        if (dictionary.TryGetValue(key, out var count)) dictionary[key] = count + 1;
        else dictionary[key] = 1;
    }
    public bool Remove(K key)
    {
        var dictionary = _dictionary;
        if (!dictionary.TryGetValue(key, out var count)) return false;
        if (count == 1) dictionary.Remove(key);
        else dictionary[key] = count - 1;
        return true;
    }
    public int this[K key] => _dictionary.TryGetValue(key, out var count) ? count : 0;
    public int CountKeys => _dictionary.Count;
}
0