-
Notifications
You must be signed in to change notification settings - Fork 0
/
IITWPC4I.cpp
151 lines (118 loc) · 2.26 KB
/
IITWPC4I.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
#include<bits/stdc++.h>
using namespace std;
#define vi vector<long long int>
#define set1 set<long long int>
#define pb push_back
#define pob pop_back
#define map1 map<long long int,long long int>
#define ll long long int
#define mp make_pair
#define inf 1e17
#define minf INT_MIN
#define setbits(x) __builtin_popcountll(x)
#define pres(x,y) fixed<<setprecision(y)<<x
#define pair1 pair<long long int,long long int>
#define mk(arr,n,type) type *arr=new type[n];
#define mod 1000000007
#define pi 3.1415926535897932384626433832
#define ins insert
#define what_is(x) cerr << #x << " is " << x << endl;
#define w(t) long long int t;cin>>t;while(t--)
struct edge
{
ll a, b, w;
};
ll id;
vector<edge> v;
vi par;
unordered_set<ll> s;
bool comp(edge e1, edge e2) {
if (e1.a == e2.a && e1.b == e2.b) {
return e1.w <= e2.w;
}
return e1.w < e2.w;
}
ll getp(ll x) {
if (par[x] == -1) {
return x;
}
return par[x] = getp(par[x]);
}
void unite(ll x, ll y) {
x = getp(x);
y = getp(y);
par[x] = y;
}
ll find_set(ll x) {
if (x == par[x])
return x;
return find_set(par[x]);
}
void solve() {
ll n, m;
cin >> n >> m;
s.clear();
par.clear();
par.assign(n + 1, -1);
v.clear();
ll res = 0;
set1 d;
for (ll i = 1; i <= n; i++) {
ll x;
cin >> x;
if (x == 1) {
id = i;
s.ins(i);
}
else {
d.ins(i);
}
}
for (auto x : s) {
if (x != id) {
v.pb({id, x, 0});
}
}
for (ll i = 0; i < m; i++) {
ll x, y, z;
cin >> x >> y >> z;
v.pb({x, y, z});
}
if (s.size() == n) {
cout << 0 << "\n";
return;
}
if (s.size() == 0) {
cout << "impossible" << "\n";
return;
}
sort(v.begin(), v.end(), comp);
for (auto x : v) {
if (getp(x.a) != getp(x.b)) {
unite(x.a, x.b);
res += x.w;
}
}
set1 f;
for (ll i = 1; i <= n; i++) {
f.ins(getp(i));
}
if (f.size() != 1) {
cout << "impossible" << "\n";
return;
}
/**/
cout << res << "\n";
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
#ifndef ONLINE_JUDGE
freopen("inp.txt", "r", stdin);
freopen("out.txt", "w", stdout);
#endif
w(t) {
solve();
}
return 0;
}