結果
問題 | 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");}}