白俊アルゴリズム14621号:私だけじゃダメな恋
16889 ワード
リンク
https://www.acmicpc.net/problem/14621
質問する
ジェイミーは24歳の母胎の独身者だ.ジェイミーは大麻法師になれないと言って、自分のプログラミング能力を利用して会議アプリケーションを作ることにした.合コンアプリは大学生を目指して大学間の道路データを収集しています.
このアプリケーションはユーザーに私心の道を提供します.このパスには3つの特徴があります.
1.私心経路はユーザーの私心を満たすためであり、南超大学と余超大学を結ぶ道しかない.
2.ユーザーが様々な人に会うために、どの大学からどの大学に移動してもよい.
3.会議の時間を無駄にしないために、このパスの長さは最短距離である必要があります.
道路データが左図、右図の紫色線で構成された経路であれば、上記3つの条件を満たす経路を作成することができる.
このとき,与えられた距離データを用いて私心経路の長さを求める.
入力
入力された1行目には、学校数Nと学校を結ぶ道路数Mが与えられる.(2 ≤ N ≤ 1,000) (1 ≤ M ≤ 10,000)
2列目の各学校が南超大学ならM、余超大学ならWです.
次のM行にはuvdがあり、u学校とv学校がつながっていることを示しています.この距離はdです.(1 ≤ u, v ≤ N) , (1 ≤ d ≤ 1,000)
しゅつりょく
Knockamiが作成したアプリケーションのパス長を出力します.(すべての学校に接続するパスがない場合は-1を出力します.)
入力と出力の例
プールコード(C++)
#include <bits/stdc++.h>
#define X first
#define Y second
#define pb push_back
#define sz(a) int((a).size())
#define fastio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
#define MAX(a, b) (((a) > (b)) ? (a) : (b))
#define all(v) v.begin(), v.end()
using namespace std;
using ll = long long;
using ull = unsigned long long;
using dbl = double;
using ldb = long double;
using pii = pair<int,int>;
using pll = pair<ll,ll>;
using vi = vector<int>;
using wector = vector<vector<int>>;
using tii = tuple<int,int,int>;
struct UnionFind {
vector<int> parent, rank, cnt;
UnionFind(int n) : parent(n), rank(n, 1), cnt(n, 1) {
iota(parent.begin(), parent.end(), 0);
}
int Find(int x) {
return x == parent[x] ? x : parent[x] = Find(parent[x]);
}
bool Union(int a, int b) {
a = Find(a), b = Find(b);
if (a == b) return 0;
if (rank[a] < rank[b]) swap(a, b);
parent[b] = a;
rank[a] += rank[a] == rank[b];
cnt[a] += cnt[b];
return 1;
}
};
int main() {
fastio;
int n,m; cin >> n >> m;
vector<char> gender(n);
UnionFind UF(n + 1);
for(int i = 0; i < n; i++){
char c; cin >> c;
gender[i] = c;
}
vector<tii> v;
for(int i = 0; i < m; i++){
int a,b,w; cin >> a >> b >> w;
if(gender[a - 1] != gender[b - 1]) v.pb({w,a,b});
}
sort(all(v));
int cost = 0,cnt = 0;
for(auto [c,a,b] : v){
a = UF.Find(a),b = UF.Find(b);
if(a == b) continue;
UF.Union(a, b);
cost += c;
if(++cnt == n - 1) break;
}
cout << (cnt != n - 1 ? -1 : cost) << "\n";
return 0;
}
重要なのは、性別の異なる大学だけが参加できるため、幹線に関する情報を入力する際に性別が同じであれば追加しないことです.出力時にcntが幹線数n−1に等しくなければ−1を出力する.
Reference
この問題について(白俊アルゴリズム14621号:私だけじゃダメな恋), 我々は、より多くの情報をここで見つけました https://velog.io/@inwooleeme/백준-알고리즘-14621번-나만-안되는-연애テキストは自由に共有またはコピーできます。ただし、このドキュメントのURLは参考URLとして残しておいてください。
Collection and Share based on the CC Protocol