結果

問題 No.1409 Simple Math in yukicoder
コンテスト
ユーザー kakel-san
提出日時 2024-07-15 23:22:30
言語 C#
(.NET 8.0.404)
結果
WA  
実行時間 -
コード長 2,816 bytes
コンパイル時間 7,298 ms
コンパイル使用メモリ 166,488 KB
実行使用メモリ 197,080 KB
最終ジャッジ日時 2024-07-15 23:22:58
合計ジャッジ時間 22,784 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 1
other AC * 1 WA * 57
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (86 ms)。
MSBuild のバージョン 17.9.6+a4ecab324 (.NET)
  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;

class Program
{
    static int NN => int.Parse(ReadLine());
    static int[] NList => ReadLine().Split().Select(int.Parse).ToArray();
    static int[][] NArr(long n) => Enumerable.Repeat(0, (int)n).Select(_ => NList).ToArray();
    static int[] LList(long n) => Enumerable.Repeat(0, (int)n).Select(_ => int.Parse(ReadLine())).ToArray();
    public static void Main()
    {
        Solve();
    }
    static void Solve()
    {
        var r = new Random();
        var t = NN;
        var ans = new int[t][];
        for (var u = 0; u < t; ++u)
        {
            var c = NList;
            var (v, x) = (c[0], c[1]);
            ans[u] = Math(r, v, x);
        }
        WriteLine(string.Join("\n", ans.Select(ai => string.Join(" ", ai))));
    }
    static int[] Math(Random r, int v, int x)
    {
        if (x == 1) return new int[] { 1 };
        if (v == 1)
        {
            var _ans = new int[x];
            for (var i = 0; i < x; ++i) _ans[i] = i + 1;
            return _ans;
        }
        var p = PDiv(v);
        foreach (var kv in PDiv(x)) if (!p.ContainsKey(kv.Key)) p[kv.Key] = kv.Value;
        var b = 0;
        while (true)
        {
            b = r.Next(v * x) + 1;
            var flg = true;
            foreach (var kv in p)
            {
                if (Exp(b, kv.Key, v * x + 1) == 1)
                {
                    flg = false;
                    break;
                }
            }
            if (flg) break;
        }
        var ans = new int[x];
        for (var i = 0; i < x; ++i)
        {
            ans[i] = (int)Exp(b, i * v, v * x + 1);
        }
        Array.Sort(ans);
        return ans;
    }
    static Dictionary<long, long> PDiv(long n)
    {
        var dic = new Dictionary<long, long>();
        var tmp = n;
        while (tmp % 2 == 0)
        {
            tmp /= 2;
            if (dic.ContainsKey(2)) ++dic[2];
            else dic.Add(2, 1);
        }
        for (var p = 3L; p * p <= n; p += 2)
        {
            while (tmp % p == 0)
            {
                tmp /= p;
                if (dic.ContainsKey(p)) ++dic[p];
                else dic.Add(p, 1);
            }
            if (tmp == 1) break;
        }
        if (tmp > 1) dic.Add(tmp, 1);
        return dic;
    }
    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