結果
問題 | No.1767 BLUE to RED |
ユーザー |
![]() |
提出日時 | 2021-11-26 22:42:28 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 276 ms / 2,000 ms |
コード長 | 2,779 bytes |
コンパイル時間 | 1,437 ms |
コンパイル使用メモリ | 128,792 KB |
最終ジャッジ日時 | 2025-01-26 01:35:05 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
#include <iostream> #include <algorithm> #include <iomanip> #include <vector> #include <queue> #include <deque> #include <set> #include <map> #include <tuple> #include <cmath> #include <numeric> #include <functional> #include <cassert> #define debug_value(x) cerr << "line" << __LINE__ << ":<" << __func__ << ">:" << #x << "=" << x << endl; #define debug(x) cerr << "line" << __LINE__ << ":<" << __func__ << ">:" << x << endl; template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; } template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; } using namespace std; typedef long long ll; template<typename T> vector<vector<T>> vec2d(int n, int m, T v){ return vector<vector<T>>(n, vector<T>(m, v)); } template<typename T> vector<vector<vector<T>>> vec3d(int n, int m, int k, T v){ return vector<vector<vector<T>>>(n, vector<vector<T>>(m, vector<T>(k, v))); } template<typename T> void print_vector(vector<T> v, char delimiter=' '){ if(v.empty()) { cout << endl; return; } for(int i = 0; i+1 < v.size(); i++) cout << v[i] << delimiter; cout << v.back() << endl; } using P = pair<int, int>; int RED = 0, BLUE = 1; struct UnionFind { vector<int> data; UnionFind(int size) : data(size, -1) {} bool unionSet(int x, int y) { x = root(x); y = root(y); if (x != y) { if (data[y] < data[x]) swap(x, y); data[x] += data[y]; data[y] = x; } return x != y; } bool findSet(int x, int y) { return root(x) == root(y); } int root(int x) { return data[x] < 0 ? x : data[x] = root(data[x]); } int size(int x) { return -data[root(x)]; } }; using T = tuple<ll, int, int>; int main(){ ios::sync_with_stdio(false); cin.tie(0); cout << setprecision(10) << fixed; int n, m; cin >> n >> m; vector<P> p; for(int i = 0; i < n; i++){ int a; cin >> a; p.push_back(P(a, RED)); } for(int i = 0; i < m; i++){ int b; cin >> b; p.push_back(P(b, BLUE)); } sort(p.begin(), p.end()); int root = 0; priority_queue<T, vector<T>, greater<T>> t; for(int i = 0; i < n+m; i++){ auto [x, col] = p[i]; int idx = i+1; if(col == RED) t.push(T(0, root, idx)); if(i > 0){ auto [x_pre, _] = p[i-1]; t.push(T(x-x_pre, idx-1, idx)); } } ll ans = 0; UnionFind uf(n+m+1); while(uf.size(root) != n+m+1){ auto [w, i, j] = t.top(); t.pop(); if(uf.findSet(i, j)) continue; // if(i != 0) cout << p[i-1].first << ' ' << p[j-1].first << endl; ans += w; uf.unionSet(i, j); } cout << ans << endl; }