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

• 問題: No.2818 A Game I Play to Pass the Time
• ユーザー: Caiiiiiiii
• 提出日時: 2024-08-18 15:10:56
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 524 ms / 2,000 ms
• コード長: 775 bytes
• コンパイル時間: 2,055 ms
• コンパイル使用メモリ: 168,296 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-08-18 15:11:13
• 合計ジャッジ時間: 15,327 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	516 ms; 5,376 KB
testcase_02	AC	495 ms; 5,376 KB
testcase_03	AC	517 ms; 5,376 KB
testcase_04	AC	498 ms; 5,376 KB
testcase_05	AC	494 ms; 5,376 KB
testcase_06	AC	518 ms; 5,376 KB
testcase_07	AC	495 ms; 5,376 KB
testcase_08	AC	520 ms; 5,376 KB
testcase_09	AC	492 ms; 5,376 KB
testcase_10	AC	512 ms; 5,376 KB
testcase_11	AC	492 ms; 5,376 KB
testcase_12	AC	490 ms; 5,376 KB
testcase_13	AC	515 ms; 5,376 KB
testcase_14	AC	489 ms; 5,376 KB
testcase_15	AC	524 ms; 5,376 KB
testcase_16	AC	491 ms; 5,376 KB
testcase_17	AC	497 ms; 5,376 KB
testcase_18	AC	513 ms; 5,376 KB
testcase_19	AC	502 ms; 5,376 KB

ソースコード

#include <bits/stdc++.h>

using LL = long long;

const int N = 67;
const int MOD = 998244353;

int inv[N];

void solve() {
  LL n, s;
  scanf("%lld%lld", &n, &s);
  int ans = s % MOD;
  for(int i = 2; i <= 100; ++i)
    if(n % i == 0) {
      LL mul = 1;
      int tmp = 0, cnt = 0;
      while(n % i == 0) {
	++cnt;
	n /= i;
	mul *= i;
      }
      for(int j = 0, c = 1; j <= cnt; ++j) {
	tmp = (tmp + mul % MOD * c) % MOD;
	c = 1LL * c * ((s + j - 1) % MOD) % MOD * inv[j + 1] % MOD;
	mul /= i;
      }
      ans = 1LL * ans * tmp % MOD;
    }
  printf("%d\n", ans);
}

int main() {

  inv[1] = 1;
  for(int i = 2; i < N; ++i)
    inv[i] = MOD - 1LL * MOD / i * inv[MOD % i] % MOD;
  
  int cases = 1;
  scanf("%d", &cases);

  while(cases--)
    solve();

  return 0;

}