제출 #873799

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
873799		AmirAli_H1	Reconstruction Project (JOI22_reconstruction)	C++17	31 / 100	4639 ms	437808 KiB

reconstruction

// In the name of Allah

#include <bits/stdc++.h>
using namespace std;
 
typedef 	long long int			ll;
typedef 	long double				ld;
typedef 	pair<int, int>			pii;
typedef 	pair<ll, ll>			pll;
 
#define 	all(x)					(x).begin(),(x).end()
#define 	len(x)					((ll) (x).size())
#define 	F						first
#define 	S						second
#define 	pb						push_back
#define 	sep						' '
#define 	endl					'\n'
#define 	Mp						make_pair
#define 	kill(x)					cout << x << '\n', exit(0)
#define 	set_dec(x)				cout << fixed << setprecision(x);
#define 	file_io(x,y)			freopen(x, "r", stdin); freopen(y, "w", stdout);
 
int n, m, q;
const int maxn = 1e5 + 4;
const int maxm = 500 + 4;
const int maxs = 1e6 + 4;
const int oo = 1e9 + 3;
int p[maxn], sz[maxn]; pii ind[maxn];
int E1[maxn][maxm], s1[maxn];
int E2[maxn][maxm], s2[maxn];
vector<int> A[maxn]; vector<int> arr, arrx, Q;
vector<pair<pii, int>> E; vector<int> vc;
pll res[maxs];

int GI(int x) {
	return lower_bound(all(arr), x) - arr.begin();
}

int GIx(int x) {
	return lower_bound(all(arrx), x) - arrx.begin();
}

int get(int a) {
	return (p[a] == a) ? a : p[a] = get(p[a]);
}
 
bool merge(int a, int b) {
	a = get(a); b = get(b);
	if (a == b) return 0;
	if (sz[a] > sz[b]) swap(a, b);
	p[a] = b; sz[b] += sz[a]; sz[a] = 0;
	return 1;
}

bool cmp(int r1, int r2, int x) {
	int w1 = abs(x - arr[E[r1].S]), w2 = abs(x - arr[E[r2].S]);
	return (Mp(w1, r1) < Mp(w2, r2));
}

void cal(ll x, int j) {
	vc.clear();
	iota(p, p + n, 0); fill(sz, sz + n, 1);
	
	int p1 = 0, p2 = 0;
	while (p1 < s1[j - 1] && p2 < s2[j]) {
		int r1 = E1[j - 1][p1], r2 = E2[j][p2];
		int w1 = abs(x - arr[E[r1].S]), w2 = abs(x - arr[E[r2].S]);
		if (cmp(r1, r2, x)) {
			if (merge(E[r1].F.F, E[r1].F.S)) vc.pb(r1);
			p1++;
		}
		else {
			if (merge(E[r2].F.F, E[r2].F.S)) vc.pb(r2);
			p2++;
		}
	}
	while (p1 < s1[j - 1]) {
		int r1 = E1[j - 1][p1]; ll w1 = x - arr[E[r1].S];
		if (merge(E[r1].F.F, E[r1].F.S)) vc.pb(r1);
		p1++;
	}
	while (p2 < s2[j]) {
		int r2 = E2[j][p2]; ll w2 = arr[E[r2].S] - x;
		if (merge(E[r2].F.F, E[r2].F.S)) vc.pb(r2);
		p2++;
	}
}

void add_val(int l, int r, pll x) {
	if (r <= l) return ;
	
	l = GIx(l); r = GIx(r);
	res[l].F += x.F; res[l].S += x.S;
	res[r].F -= x.F; res[r].S -= x.S;
}

void cal1(int ind, int i) {
	iota(p, p + n, 0); fill(sz, sz + n, 1);
	
	int ux = E[ind].F.F, vx = E[ind].F.S; ll wx = arr[E[ind].S];
	int j = find(E2[i], E2[i] + s2[i], ind) - E2[i];
	for (int r = 0; r < j; r++) {
		int x = E2[i][r];
		int u = E[x].F.F, v = E[x].F.S;
		merge(u, v);
	}
	for (int r = 0; r < s1[i - 1]; r++) {
		int x = E1[i - 1][r];
		int u = E[x].F.F, v = E[x].F.S;
		if ((get(u) == get(ux) && get(v) == get(vx)) || (get(u) == get(vx) && get(v) == get(ux))) {
			int w = arr[E[x].S];
			int R = (w + wx) / 2;
			if (!cmp(ind, x, R)) R++;
			add_val(R, wx + 1, Mp(-1, wx));
			return ;
		}
		merge(u, v);
	}
	add_val(-oo, wx + 1, Mp(-1, wx));
}

void cal2(int ind, int i) {
	iota(p, p + n, 0); fill(sz, sz + n, 1);
	
	int ux = E[ind].F.F, vx = E[ind].F.S; ll wx = arr[E[ind].S];
	int j = find(E1[i], E1[i] + s1[i], ind) - E1[i];
	for (int r = 0; r < j; r++) {
		int x = E1[i][r];
		int u = E[x].F.F, v = E[x].F.S;
		merge(u, v);
	}
	for (int r = 0; r < s2[i + 1]; r++) {
		int x = E2[i + 1][r];
		int u = E[x].F.F, v = E[x].F.S;
		if ((get(u) == get(ux) && get(v) == get(vx)) || (get(u) == get(vx) && get(v) == get(ux))) {
			int w = arr[E[x].S];
			int R = (w + wx) / 2;
			if (!cmp(ind, x, R)) R--;
			add_val(wx, R + 1, Mp(1, -wx));
			return ;
		}
		merge(u, v);
	}
	add_val(wx, oo, Mp(1, -wx));
}

int main() {
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	
	cin >> n >> m;
	for (int i = 0; i < m; i++) {
		int u, v, w;
		cin >> u >> v >> w; u--; v--;
		E.pb(Mp(Mp(u, v), w)); arr.pb(w);
	}
	cin >> q;
	for (int i = 0; i < q; i++) {
		int x; cin >> x;
		Q.pb(x); arrx.pb(x);
	}
	
	arr.pb(-oo); arr.pb(oo);
	arrx.pb(-oo); arrx.pb(oo);
	sort(all(arr)); arr.resize(unique(all(arr)) - arr.begin());
	sort(all(arrx)); arrx.resize(unique(all(arrx)) - arrx.begin());
	
	for (int i = 0; i < m; i++) {
		E[i].S = GI(E[i].S);
		A[E[i].S].pb(i);
	}
	
	for (int i = 0; i < len(arr); i++) {
		iota(p, p + n, 0); fill(sz, sz + n, 1);
		for (int j : A[i]) {
			int u = E[j].F.F, v = E[j].F.S;
			if (merge(u, v)) E1[i][s1[i]++] = j;
		}
		if (i - 1 >= 0) {
			for (int r = 0; r < s1[i - 1]; r++) {
				int j = E1[i - 1][r];
				int u = E[j].F.F, v = E[j].F.S;
				if (merge(u, v)) E1[i][s1[i]++] = j;
			}
		}
	}
	for (int i = len(arr) - 1; i >= 0; i--) {
		iota(p, p + n, 0); fill(sz, sz + n, 1);
		for (int j : A[i]) {
			int u = E[j].F.F, v = E[j].F.S;
			if (merge(u, v)) E2[i][s2[i]++] = j;
		}
		if (i + 1 < len(arr)) {
			for (int r = 0; r < s2[i + 1]; r++) {
				int j = E2[i + 1][r];
				int u = E[j].F.F, v = E[j].F.S;
				if (merge(u, v)) E2[i][s2[i]++] = j;
			}
		}
	}
	
	fill(ind, ind + m, Mp(oo, -oo));
	for (int i = 1; i < len(arr) - 1; i++) {
		cal(arr[i], i);
		for (int j : vc) {
			ind[j].F = min(ind[j].F, i);
			ind[j].S = max(ind[j].S, i);
		}
	}
	
	for (int i = 0; i < m; i++) {
		if (ind[i].F == oo) continue;
		
		int i1 = ind[i].F, i2 = ind[i].S;
		if (i < 50000) {
			cal1(i, i1); cal2(i, i2);
		}
	}
	
	for (int i = 1; i <= len(arrx); i++) {
		res[i].F += res[i - 1].F; res[i].S += res[i - 1].S;
	}
	for (int i = 0; i < q; i++) {
		ll x = Q[i]; int j = GIx(x);
		cout << res[j].F * x + res[j].S << endl;
	}
	
	return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

reconstruction.cpp: In function 'void cal(ll, int)':
reconstruction.cpp:67:7: warning: unused variable 'w1' [-Wunused-variable]
   67 |   int w1 = abs(x - arr[E[r1].S]), w2 = abs(x - arr[E[r2].S]);
      |       ^~
reconstruction.cpp:67:35: warning: unused variable 'w2' [-Wunused-variable]
   67 |   int w1 = abs(x - arr[E[r1].S]), w2 = abs(x - arr[E[r2].S]);
      |                                   ^~
reconstruction.cpp:78:30: warning: unused variable 'w1' [-Wunused-variable]
   78 |   int r1 = E1[j - 1][p1]; ll w1 = x - arr[E[r1].S];
      |                              ^~
reconstruction.cpp:83:26: warning: unused variable 'w2' [-Wunused-variable]
   83 |   int r2 = E2[j][p2]; ll w2 = arr[E[r2].S] - x;
      |                          ^~

• Subtask 1: 3 / 3
• Subtask 2: 0 / 4
• Subtask 3: 0 / 7
• Subtask 4: 28 / 28
• Subtask 5: 0 / 35
• Subtask 6: 0 / 23