結果

問題 No.1409 Simple Math in yukicoder
コンテスト
ユーザー Nachia
提出日時 2021-02-26 23:23:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 28 ms / 2,000 ms
コード長 1,045 bytes
コンパイル時間 3,196 ms
コンパイル使用メモリ 202,540 KB
最終ジャッジ日時 2025-01-19 07:12:46
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 58
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:38:15: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   38 |   int T; scanf("%d",&T);
      |          ~~~~~^~~~~~~~~
main.cpp:40:19: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   40 |     ULL v,x; scanf("%llu%llu",&v,&x);
      |              ~~~~~^~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using LL=long long;
using ULL=unsigned long long;
#define rep(i,n) for(int i=0; i<(n); i++)

vector<pair<ULL,ULL>> factorize(ULL N){
 vector<pair<ULL,ULL>> res;
 for(ULL i=2; i*i<=N; i++){
  if(N%i) continue;
  res.push_back({i,0});
  while(N%i==0){ N/=i; res.back().second++; }
 }
 if(N!=1) res.push_back({N,1});
 return move(res);
}

ULL powm(ULL a,ULL i,ULL m){
  if(i==0) return 1;
  ULL r=powm(a*a%m,i/2,m);
  if(i%2==1) r=r*a%m;
  return r;
}
ULL findg(ULL v,ULL x){
  ULL m=x*v+1;
  auto F=factorize(x*v);
  ULL g=2;
  while(true){
    bool ok=true;
    for(auto f:F) if(powm(g,x*v/f.first,m)==1) ok=false;
    if(ok) break;
    g++;
  }
  return powm(g,v,m);
}

int main(){
  int T; scanf("%d",&T);
  while(T--){
    ULL v,x; scanf("%llu%llu",&v,&x);
    ULL m=v*x+1;
    ULL g=findg(v,x);
    vector<ULL> ans={1}; rep(i,x-1) ans.push_back(ans.back()*g%m);
    sort(ans.begin(),ans.end());
    rep(i,x){ if(i) printf(" "); printf("%u",(unsigned int)ans[i]); } printf("\n");
  }
  return 0;
}
0