結果

問題 No.1409 Simple Math in yukicoder
コンテスト
ユーザー persimmon-persimmon
提出日時 2021-03-02 13:31:39
言語 PyPy3
(7.3.23 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
実行時間 -
コード長 591 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 86 ms
コンパイル使用メモリ 80,384 KB
実行使用メモリ 326,152 KB
最終ジャッジ日時 2026-09-09 08:22:24
合計ジャッジ時間 38,838 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 35 TLE * 1 -- * 22
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from random import randint
def main(u,x):
  p=x*u+1
  if u==1:return list(range(1,p))
  if x==1:return [x]
  # p:prime
  # a^(p-1)=1 mod p
  # p=x*u+1
  # a^(x*u)=1 mod p
  # (a^u)^x=1 mod p

  while True:
    i=randint(2,p-1)
    s={i}
    ii=i*i%p
    while ii!=i:
      s.add(ii)
      ii=ii*i%p
    if len(s)==p-1:break
  i=pow(i,u,p)
  ary=[]
  ii=i
  for _ in range(x):
    ary.append(ii)
    ii=ii*i%p
  ary.sort()
  return ary
if __name__=='__main__':
  t=int(input())
  cases=[list(map(int,input().split())) for _ in range(t)]
  for u,x in cases:
    ret=main(u,x)
    print(*ret)
0