結果

問題 No.1409 Simple Math in yukicoder
コンテスト
ユーザー publfl
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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");
	}
}
0