-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathgoat.cpp
98 lines (82 loc) · 1.82 KB
/
goat.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
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pi = pair<ll, ll>;
using vi = vector<ll>;
using vpi = vector<pi>;
using vb = vector<bool>;
struct DSU {
vi par;
vi sz;
ll tot;
void init(ll n) {
for (ll i = 0; i < n; i++) {
par.push_back(i);
sz.push_back(1);
tot = 0;
}
}
ll get_rep(ll node) {
if (par[node] == node) return node;
return get_rep(par[node]);
}
ll get_sz(ll node) {
return sz[get_rep(node)];
}
bool check_con(ll a, ll b) {
return get_rep(a) == get_rep(b);
}
void unite(ll a, ll b) {
ll x = get_rep(a);
ll y = get_rep(b);
if (x == y) return;
if (sz[x] < sz[y]) {
par[x] = y;
tot -= sz[x] / 2;
tot -= sz[y] / 2;
sz[y] += sz[x];
tot += sz[y] / 2;
} else {
par[y] = x;
tot -= sz[x] / 2;
tot -= sz[y] / 2;
sz[x] += sz[y];
tot += sz[x] / 2;
}
}
};
void solve() {
ll n;
cin >> n;
vi a(n);
for (int i = 0; i < n; i++) {
cin >> a[i];
}
sort(a.begin(), a.end());
priority_queue<pi> q;
DSU dsu;
dsu.init(n);
ll best = n;
for (int i = n - 1; i >= 0; i--) {
ll ans = i;
while (!q.empty()) {
if (q.top().first <= a[i]) break;
pi top = q.top();
q.pop();
ll v = top.second;
dsu.unite(v, v + 1);
}
if(i > 0) q.push({2 * a[i - 1] - a[i], i-1});
ans += dsu.tot;
best = min(best, ans);
}
cout << best << endl;
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
ll t;
cin >> t;
while (t--) solve();
return 0;
}