#991979 (C++23) No.2818 A Game I Play to Pass the Time

• 問題: No.2818 A Game I Play to Pass the Time
• ユーザー: highlighter
• 提出日時: 2024-06-29 13:35:57
• 言語: C++23 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 414 ms / 2,000 ms
• コード長: 967 bytes
• コンパイル時間: 5,415 ms
• コンパイル使用メモリ: 312,340 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-06-30 15:10:02
• 合計ジャッジ時間: 16,997 ms
• ジャッジサーバーID （参考情報）: judge4 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 6,816 KB
testcase_01	AC	414 ms; 6,816 KB
testcase_02	AC	411 ms; 6,944 KB
testcase_03	AC	410 ms; 6,940 KB
testcase_04	AC	411 ms; 6,940 KB
testcase_05	AC	411 ms; 6,940 KB
testcase_06	AC	410 ms; 6,940 KB
testcase_07	AC	412 ms; 6,944 KB
testcase_08	AC	409 ms; 6,944 KB
testcase_09	AC	409 ms; 6,940 KB
testcase_10	AC	408 ms; 6,944 KB
testcase_11	AC	411 ms; 6,940 KB
testcase_12	AC	411 ms; 6,944 KB
testcase_13	AC	408 ms; 6,940 KB
testcase_14	AC	410 ms; 6,940 KB
testcase_15	AC	411 ms; 6,940 KB
testcase_16	AC	410 ms; 6,944 KB
testcase_17	AC	410 ms; 6,940 KB
testcase_18	AC	404 ms; 6,944 KB
testcase_19	AC	408 ms; 6,944 KB

ソースコード

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
using mint=modint998244353;

int prime[25]={2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67,71,73,79,83,89,97};

int main(){
	int Q;
	scanf("%d",&Q);
	mint inv[61];
	inv[1]=1;
	for(int i=2;i<61;i++){
		inv[i]=-inv[998244353%i]*(998244353/i);
	}
	for(;Q--;){
		long long N,S;
		scanf("%lld%lld",&N,&S);
		vector<pair<int,int>> vec;
		for(int i=0;i<25;i++){
			if(N%prime[i]){
				continue;
			}
			int count=0;
			while(N%prime[i]==0){
				N/=prime[i];
				count++;
			}
			vec.emplace_back(make_pair(prime[i],count));
		}
		mint ans=1;
		for(auto p : vec){
			vector<mint> data(p.second+1);
			data[0]=1;
			for(int i=1;i<=p.second;i++){
				data[i]=data[i-1]*p.first;
			}
			mint res=data[p.second];
			mint mem=1;
			for(int i=1;i<=p.second;i++){
				mem*=inv[i]*(S-2+i);
				res+=data[p.second-i]*mem;
			}
			ans*=res;
		}
		ans*=S;
		printf("%d\n",ans.val());
	}
}