結果
問題 |
No.3087 University Coloring
|
ユーザー |
![]() |
提出日時 | 2025-04-04 21:35:13 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 89 ms / 2,000 ms |
コード長 | 3,944 bytes |
コンパイル時間 | 2,573 ms |
コンパイル使用メモリ | 208,752 KB |
実行使用メモリ | 8,732 KB |
最終ジャッジ日時 | 2025-04-04 21:37:48 |
合計ジャッジ時間 | 7,083 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
//#pragma GCC target("avx2") //#pragma GCC optimize("O3") //#pragma GCC optimize("unroll-loops") #include <bits/stdc++.h> using namespace std; using ll = long long; using pii = pair<int,int>; using pll = pair<ll,ll>; using pli = pair<ll,int>; #define TEST cerr << "TEST" << endl #define AMARI 998244353 //#define AMARI 1000000007 #define el '\n' #define El '\n' #define YESNO(x) ((x) ? "Yes" : "No") #define YES YESNO(true) #define NO YESNO(false) #define REV_PRIORITY_QUEUE(tp) priority_queue<tp,vector<tp>,greater<tp>> #define NO_ANS(x) {cout << (x) << '\n'; return;} template <typename T> void inline VEC_UNIQ(vector<T> &v){ sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end()); return; } //UnionFind T には何か乗せたいものを乗せられる(連結成分の大きさなど) template <typename T> class ococo_unionfind { private: vector<pair<int,int>> g; vector<T> vec; int com_cnt; T func(T a,T b){ return (a + b); } public: ococo_unionfind(int n = 0){ g.resize(n); vec.resize(n); for(int i = 0; i < n; i++)g[i] = pair(i,1); com_cnt = n; } ococo_unionfind(vector<T> v){ int n = (int)v.size(); g.resize(n); vec = v; for(int i = 0; i < n; i++)g[i] = pair(i,1); com_cnt = n; } int ne(int idx){ if(g[idx].first == idx)return idx; else return g[idx].first = ne(g[idx].first); } //(a,b) 間をマージする。既に連結されていたら true,そうでなければ false を返す bool einsert(int a,int b){ a = ne(a); b = ne(b); if(a == b)return true; com_cnt--; vec[a] = vec[b] = func(vec[a],vec[b]); if(g[a].second < g[b].second){ g[a].first = b; g[b].second = max(g[a].second + 1,g[b].second); } else{ g[b].first = a; g[a].second = max(g[b].second + 1,g[a].second); } return false; } bool peinsert(pair<int,int> p){ return einsert(p.first,p.second); } bool is_connect(int a,int b){ return (ne(a) == ne(b)); } bool pis_connect(pair<int,int> p){ return is_connect(p.first,p.second); } //いくつの連結成分に分かれているか出力する int component_count(void){ return com_cnt; } //連結成分に乗せたものを返す T get_val(int a){ return vec[ne(a)]; } }; #define MULTI_TEST_CASE false void solve(void){ //問題を見たらまず「この問題設定から言えること」をいっぱい言う //一個回答に繋がりそうな解法が見えても、実装や細かい詰めに時間がかかりそうなら別の方針を考えてみる //ある程度考察しても全然取っ掛かりが見えないときは実験をしてみる //よりシンプルな問題に言い換えられたら、言い換えた先の問題を自然言語ではっきりと書く //複数の解法のアイデアを思いついた時は全部メモしておく //g++ -D_GLIBCXX_DEBUG -Wall -Wextra -O2 e.cpp -o o int n; cin >> n; ococo_unionfind<int> uf(n); int m; cin >> m; vector<pair<ll,pii>> edge(m); for(int i = 0; i < m; i++){ int u,v; cin >> u >> v; u--; v--; ll w; cin >> w; edge[i] = pair(w,pair(u,v)); } sort(edge.rbegin(),edge.rend()); ll ans = 0LL; for(int i = 0; i < m; i++){ ll w; pii temp; tie(w,temp) = edge[i]; if(uf.pis_connect(temp))continue; uf.peinsert(temp); ans += w; } cout << ans * 2LL << el; return; } void calc(void){ return; } signed main(void){ cin.tie(nullptr); ios::sync_with_stdio(false); calc(); int t = 1; if(MULTI_TEST_CASE)cin >> t; while(t--){ solve(); } return 0; }