This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#pragma GCC optimize ("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,popcnt,lzcnt")
//besmellah
#include <bits/stdc++.h>
#define FOR(i, a, b) for (int i = (a); i < (b); ++i)
#define ers(v, X) adj[v].erase(find(adj[v].begin(), adj[v].end(), (X)))
using namespace std;
using ar = array<int, 2>;
using ar3 = array<int, 3>;
const int M = 1e5 + 5, N = 505, Q = 1e6 + 5;
const long long INF = 1ll << 30;
namespace dsu {
int dpr[N];
void init() {
memset(dpr, -1, sizeof dpr);
}
int gpr(int x) {
return dpr[x] < 0 ? x : dpr[x] = gpr(dpr[x]);
}
bool merge(int u, int v) {
u = gpr(u), v = gpr(v);
if (u == v) return false;
if (dpr[u] < dpr[v]) swap(u, v);
dpr[v] += dpr[u];
dpr[u] = v;
return true;
}
}
vector<ar> adj[N];
int n, m;
ar3 W[N], E[M];
bitset<N> seen;
void dfs(int v, int p) {
seen[v] = true;
for (auto [u, w]: adj[v]) if (u != p) {
W[u] = max(W[v], {w, u, v});
dfs(u, v);
}
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n >> m;
FOR(i, 0, m) {
cin >> E[i][1] >> E[i][2] >> E[i][0];
--E[i][1], --E[i][2];
if (E[i][1] > E[i][2]) swap(E[i][1], E[i][2]);
}
sort(E, E + m);
vector<ar3> undo;
deque<ar3> Es;
for (int i = m - 1; i >= 0; i--) { //ino mishe behtar kard ba DSU vali halesh nist (orderesham ye log kamtar dare)
auto [w, u, v] = E[i];
W[u] = {-INF, u, u};
seen = 0;
dfs(u, u);
if (seen[v]) {
auto [dw, du, dv] = W[v];
if (du > dv) swap(du, dv);
ers(du, (ar{dv, dw}));
ers(dv, (ar{du, dw}));
Es.erase(find(Es.begin(), Es.end(), ar3{dw, du, dv}));
//cout << "DELETING " << du + 1 << ' ' << dv + 1 << " " << dw << endl;
undo.push_back({dw, du, dv});
}
else
undo.push_back({-1, -1, -1});
//cout << "ADDING " << u + 1 << ' ' << v + 1 << ' ' << w << endl;
Es.push_back(E[i]);
adj[v].push_back({u, w});
adj[u].push_back({v, w});
}
//for (auto [w, u, v]: Es) cout << u << ' ' << v << ' ' << w << endl;
vector<ar3> jelo;
//be zaman ahamiat nemidaham :|
int q; cin >> q; int p = 0;
while (q--) {
int x; cin >> x;
bool t = false;
while (p < m && E[p][0] < x) { //ino hesesh bood vali log ziad dare
jelo.push_back(E[p]);
t = true;
Es.erase(find(Es.begin(), Es.end(), E[p]));
auto akh = undo.back();
undo.pop_back();
if (akh[0] != -1) Es.push_back(akh);
++p;
}
if (t) {
vector<ar3> ok;
ok.reserve(jelo.size());
dsu::init();
sort(jelo.rbegin(), jelo.rend());
for (auto [w, u, v]: jelo) if (dsu::merge(u, v)) ok.push_back({w, u, v});
jelo = ok;
assert(jelo.size() < N);
sort(Es.begin(), Es.end());
assert(Es.size() < N);
}
for (auto &[w, u, v]: jelo) w = x - w;
for (auto &[w, u, v]: Es) w -= x;
vector<ar3> hame(jelo.size() + Es.size());
merge(jelo.begin(), jelo.end(), Es.begin(), Es.end(), hame.begin());
dsu::init();
long long ans = 0;
//int cnt = 1;
for (auto [w, u, v]: hame) {
ans += dsu::merge(u, v) ? w : 0;
}
cout << ans << '\n';
for (auto &[w, u, v]: jelo) w = x - w;
for (auto &[w, u, v]: Es) w += x;
}
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |