結果

問題 No.3572 Number of special equations
コンテスト
ユーザー tobisatis
提出日時 2026-06-19 23:18:12
言語 C#
(.NET 10.0.201)
コンパイル:
dotnet_c
実行:
/usr/bin/dotnet_wrap
結果
WA  
実行時間 -
コード長 2,172 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 6,798 ms
コンパイル使用メモリ 173,632 KB
実行使用メモリ 75,832 KB
最終ジャッジ日時 2026-06-19 23:19:06
合計ジャッジ時間 11,007 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge1_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 6 WA * 1 TLE * 1 -- * 9
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (90 ミリ秒)。
/home/judge/data/code/Main.cs(14,12): warning CS8321: ローカル関数 'Range' は宣言されていますが、一度も使用されていません [/home/judge/data/code/main.csproj]
  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

using System.Numerics;

#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

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

long Naive(int N, int n)
{
    var res = (long)n * (n + 1) / 2 + 1;
    for (var d = 3; d <= n; d++)
    {
        var theta = Math.PI * 2 / d;
        var r = new Complex(Math.Cos(theta), Math.Sin(theta));
        var rz = new Complex[d + 1];
        rz[0] = 1;
        for (var i = 1; i < d; i++) rz[i] = rz[i - 1] * r;
        var l = n - d;
        var a = new int[l];
        var ds = new HashSet<long>();
        void S(int j, int next)
        {
            if (j == l)
            {
                var s = new HashSet<(Complex, int)>(){ (1, 0) };
                for (var i = 0; i < l; i++)
                {
                    var rot = rz[a[i]];
                    var ns = new HashSet<(Complex, int)>();
                    foreach (var k in s)
                    {
                        var (kv, kc) = k;
                        ns.Add(k);
                        ns.Add((kv * rot, kc + 1));
                    }
                    s = ns;
                }
                var ims = new double[l + 1];
                foreach (var (kv, kc) in s) ims[kc] += kv.Imaginary;
                for (var i = 0; i <= l; i++) if (Math.Abs(ims[i]) > 0.000001) return;
                var hash = 0L;
                for (var i = 0; i < l; i++) hash = hash * (N + 2) + a[i] + 1;
                ds.Add(hash);
                return;
            }
            for (var i = next; i <= d; i++)
            {
                a[j] = i;
                S(j + 1, i);
            }
        }
        S(0, 0);
        res += ds.Count;
    }
    return res;
}

// var N = 10;
// var ll = new List<long>();
// for (var n = 1; n <= N; n++)
// {
//     var res = Naive(N, n);
//     ll.Add(res);
// }
// Console.WriteLine(string.Join(",", ll));

var N = I<int>();
Console.WriteLine(Naive(N, N));
0