結果

問題 No.1917 LCMST
コンテスト
ユーザー kiyoshi0205
提出日時 2022-04-30 00:38:14
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 130 ms / 4,000 ms
コード長 920 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,778 ms
コンパイル使用メモリ 185,672 KB
実行使用メモリ 20,592 KB
最終ジャッジ日時 2026-05-10 07:15:36
合計ジャッジ時間 10,060 ms
ジャッジサーバーID
(参考情報)
judge1_0 / judge3_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#pragma GCC target("avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include<iostream>
#include<numeric>
using namespace std;
using ll=long long;
using P=tuple<ll,int,int>;
#include<atcoder/dsu>


int N;
ll ans;
bool vis[100001];
int edgecnt;
P edge[1066750];
int main(){
  atcoder::dsu uf(100001);
  cin.tie(0)->sync_with_stdio(0);
  cin>>N;
  while(N--){
    int x;
    cin>>x;
    if(vis[x])ans+=x;
    else vis[x]=true;
  }
  for(int d=1;d<=100000;++d){
    int mi=0;
    for(int k=1;d*k<=100000;++k){
      if(vis[d*k]){
        if(mi){
          auto&[c,u,v]=edge[edgecnt++];
          c=d*lcm<ll>(mi,k);
          u=d*mi;
          v=d*k;
        }else mi=k;
      }
    }
  }
  sort(edge,edge+edgecnt,[](P& l,P& r){return get<0>(l)<get<0>(r);});
  for(int i=0;i<edgecnt;++i){
    auto&[c,u,v]=edge[i];
    if(!uf.same(u,v)){
      uf.merge(u,v);
      ans+=c;
    }
  }
  cout<<ans<<'\n';
}
0