結果

問題 No.1728 [Cherry 3rd Tune] Bullet
ユーザー 👑 NachiaNachia
提出日時 2021-10-29 22:53:12
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,892 bytes
コンパイル時間 918 ms
コンパイル使用メモリ 88,720 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-07 12:32:39
合計ジャッジ時間 1,887 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

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, 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