結果
問題 | No.1409 Simple Math in yukicoder |
ユーザー |
![]() |
提出日時 | 2021-02-26 23:00:52 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 866 ms / 2,000 ms |
コード長 | 1,195 bytes |
コンパイル時間 | 480 ms |
コンパイル使用メモリ | 51,840 KB |
実行使用メモリ | 6,784 KB |
最終ジャッジ日時 | 2024-10-02 15:50:09 |
合計ジャッジ時間 | 54,172 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 58 |
ソースコード
#include <stdio.h> #include <vector> long long int power(long long int a, long long int b, int p) { long long int ans = 1; long long int k = a; while(b) { if(b%2==1) ans*=k, ans%=p; k*=k, k%=p; b/=2; } return ans; } long long int inv(long long int a, int p) { return power(a,p-2,p); } long long int fact[110]; std::vector<int> ans; long long int check[100010]; std::vector<int> save[1010][110]; int main() { fact[0] = 1; int T; scanf("%d",&T); while(T--) { ans.clear(); int a,b; scanf("%d%d",&a,&b); if(!save[a][b].empty()) { for(int i=0;i<save[a][b].size();i++) printf("%d ",save[a][b][i]); printf("\n"); continue; } int p = a*b+1; int count = 0; for(int i=1;i<=a*b;i++) check[i] = 0; for(int i=1;i<=a*b;i++) { if(check[i]==0) { long long int t = power(i,b,p); long long int i2 = i; long long int s = t; while(check[i2]==0) { check[i2] = s; if(s==1) count++; if(count>=b) goto u; i2 = (i2*i)%p; s = (s*t)%p; } } } u:; for(int i=1;i<=a*b;i++) if(check[i]==1) ans.push_back(i); save[a][b] = ans; for(int i=0;i<ans.size();i++) printf("%d ",ans[i]); printf("\n"); } }