結果

問題 No.1409 Simple Math in yukicoder
ユーザー 👑 NachiaNachia
提出日時 2021-02-26 23:11:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,100 bytes
コンパイル時間 3,284 ms
コンパイル使用メモリ 202,484 KB
最終ジャッジ日時 2025-01-19 06:55:27
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 4 WA * 30 TLE * 24
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:39:15: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   39 |   int T; scanf("%d",&T);
      |          ~~~~~^~~~~~~~~
main.cpp:41:19: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   41 |     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);
  ULL g=2;
  while(true){
    bool ok=true;
    for(auto f:F) if(powm(g,x/f.first,m)==1) ok=false;
    if(ok) break;
  }
  ULL res=g, ig=powm(g,x,m), i=ig;
  while(i!=1){ res=res*g%m; i=i*ig%m; }
  return res;
}

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