結果

問題 No.1728 [Cherry 3rd Tune] Bullet
ユーザー 👑 NachiaNachia
提出日時 2021-10-29 22:55:54
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 1,901 bytes
コンパイル時間 1,030 ms
コンパイル使用メモリ 88,028 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-07-29 08:51:58
合計ジャッジ時間 2,170 ms
ジャッジサーバーID
(参考情報)
judge11 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 1 ms
4,380 KB
testcase_06 AC 1 ms
4,380 KB
testcase_07 AC 1 ms
4,376 KB
testcase_08 AC 1 ms
4,380 KB
testcase_09 AC 2 ms
4,376 KB
testcase_10 AC 2 ms
4,376 KB
testcase_11 AC 1 ms
4,376 KB
testcase_12 AC 2 ms
4,380 KB
testcase_13 AC 2 ms
4,376 KB
testcase_14 AC 2 ms
4,376 KB
testcase_15 AC 1 ms
4,376 KB
testcase_16 AC 2 ms
4,380 KB
testcase_17 AC 2 ms
4,376 KB
testcase_18 AC 2 ms
4,376 KB
testcase_19 AC 1 ms
4,376 KB
testcase_20 AC 1 ms
4,380 KB
testcase_21 AC 2 ms
4,376 KB
testcase_22 AC 2 ms
4,380 KB
testcase_23 AC 2 ms
4,380 KB
testcase_24 AC 2 ms
4,376 KB
testcase_25 AC 4 ms
4,376 KB
testcase_26 AC 1 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
#include <atcoder/modint>
using namespace std;
using i32 = int32_t;
using u32 = uint32_t;
using i64 = int64_t;
using u64 = uint64_t;
#define rep(i,n) for(int i=0; i<(n); i++)


using m32 = atcoder::modint1000000007;

vector<pair<u64,u64>> factorize(u64 n){
  vector<pair<u64,u64>> res;
  for(u64 i=2; i*i<=n; i++){
    if(n%i) continue;
    res.push_back({i,0});
    while(n%i==0){ n/=i; res.back().second++; }
  }
  if(n!=1) res.push_back({n,1});
  return res;
}

vector<pair<u64,u64>> gcd_frequency(u64 n){
  vector<pair<u64,u64>> ans;
  ans.push_back(make_pair((u64)1, (u64)1));
  for(auto f : factorize(n)){
    vector<pair<u64,u64>> buf;
    u64 multiplier = 1;
    u64 inv_multiplier = 1;
    for(u64 p_idx=0; p_idx<f.second; p_idx++) inv_multiplier *= f.first;
    for(u64 p_idx=0; p_idx<=f.second; p_idx++){
      if(p_idx != 0){
        multiplier *= f.first;
        inv_multiplier /= f.first;
      }
      u64 freq = inv_multiplier;
      if(p_idx != f.second) freq = freq / f.first * (f.first - 1);
      for(auto a : ans) buf.push_back(make_pair(a.first*multiplier, a.second*freq));
    }
    swap(ans, buf);
  }
  return ans;
}


void testcase(){
  u64 num_vertices; cin >> num_vertices;
  i64 num_colors; cin >> num_colors;

  m32 ans = 0;
  for(auto g : gcd_frequency(num_vertices)){
    m32 simple_count = m32(num_colors).pow(g.first * 2);
    ans += simple_count * g.second;
  }
  ans /= num_vertices;
  ans += m32(num_colors).pow(num_vertices);
  ans /= 2;
  cout << ans.val() << "\n";
}


int main(){
  int testcase_count; cin >> testcase_count;
  for(int testcase_idx = 0; testcase_idx < testcase_count; testcase_idx++){
    testcase();
  }
  return 0;
}




struct ios_do_not_sync {
    ios_do_not_sync() {
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
    }
} ios_do_not_sync_instance;
0