【洛谷P2387】魔法森林

题目大意:给定一个 N 个点,M 条边的无向图,边有两个边权 a, b,求从 1 号节点到 N 号节点路径的两个权值和的最大值最小是多少。

题解:
对于有两个属性的结构的最优化问题,可以考虑先按照其中一个分量进行排序。接着从小到大枚举这个有序的分量,计算以当前枚举到的值为这一分量的最大值时,全局的最优解是多少。因此,需要高效维护的是如何求出另一个分量的最优解。

对于这道题来说,考虑对 a 分量进行排序,并按从小到大的顺序依次加边。对于即将加入的第 i 条边来说,若加入这条边使得两个本来不联通的点联通,则直接加入;若加入这条边之后,形成了环,则比较加入这条边 b 的权值和这条边两个端点之间路径上 b 的最大值,若当前边的 b 更小,则断开路径上最大边权的边,并加入当前这条边即可。利用 lct 进行维护 b 即可。

代码如下

#include <bits/stdc++.h>

using namespace std;

struct edge {
	int x, y, a, b;
	edge(int _x = 0, int _y = 0, int _a = 0, int _b = 0) {
		x = _x, y = _y;
		a = _a, b = _b;
	}
};

struct node {
	node* l;
	node* r;
	node* p;
	int rev, b, maxb, id;
	node(int _b = 0, int _id = -1) {
		l = r = p = NULL;
		b = _b;
		id = _id;
		rev = 0;
	}
	void unsafe_reverse() {
		swap(l, r);
		rev ^= 1;
	}
	void pull() {
		maxb = b;
		if (l != NULL) {
			l->p = this;
			maxb = max(maxb, l->maxb);
		}
		if (r != NULL) {
			r->p = this;
			maxb = max(maxb, r->maxb);
		}
	}
	void push() {
		if (rev) {
			if (l != NULL) {
				l->unsafe_reverse();
			} 
			if (r != NULL) {
				r->unsafe_reverse();
			}
			rev = 0;
		}
	}
};
bool is_root(node* v) {
	if (v == NULL) {
		return false;
	}
	return (v->p == NULL) || (v->p->l != v && v->p->r != v);
}
void rotate(node* v) {
	node* u = v->p;
	assert(u != NULL);
	v->p = u->p;
	if (v->p != NULL) {
		if (v->p->l == u) {
			v->p->l = v;
		}
		if (v->p->r == u) {
			v->p->r = v;
		}
	}
	if (v == u->l) {
		u->l = v->r;
		v->r = u;
	}
	if (v == u->r) {
		u->r = v->l;
		v->l = u;
	}
	u->pull();
	v->pull();
}
void deal_with_push(node* v) {
	static stack<node*> s;
	while (true) {
		s.push(v);
		if (is_root(v)) {
			break;
		}
		v = v->p;
	}
	while (!s.empty()) {
		s.top()->push();
		s.pop();
	}
}
void splay(node* v) {
	deal_with_push(v);
	while (!is_root(v)) {
		node* u = v->p;
		if (!is_root(u)) {
			if ((v == u->l) ^ (u == u->p->l)) {
				rotate(v);
			} else {
				rotate(u);
			}
		}
		rotate(v);
	}
}
void access(node* v) {
	node* u = NULL;
	while (v != NULL) {
		splay(v);
		v->r = u;
		v->pull();
		u = v;
		v = v->p;
	}
}
void make_root(node* v) {
	access(v);
	splay(v);
	v->unsafe_reverse();
}
node* find_root(node* v) {
	access(v);
	splay(v);
	while (v->l != NULL) {
		v->push();
		v = v->l;
	}
	splay(v);
	return v;
}
void link(node* v, node* u) {
	if (find_root(v) != find_root(u)) {
		make_root(v);
		v->p = u;
	}
}
void cut(node* v, node* u) {
	make_root(v);
	if (find_root(u) == v && u->p == v && u->l == NULL) {
		u->p = v->r = NULL;
		v->pull();
	}
}
void split(node* v, node* u) {
	make_root(v);
	access(u);
	splay(u);
}
node* find(node* v, int b) {
	while (true) {
		if (v->b == b) {
			break;
		}
		if (v->l != NULL && v->l->maxb == b) {
			v = v->l;
		} else {
			v = v->r;
		}
	}
	return v;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int n, m; // 0-indexed
	cin >> n >> m;
	vector<edge> e;
	for (int i = 0; i < m; i++) {
		int x, y, a, b;
		cin >> x >> y >> a >> b;
		e.emplace_back(--x, --y, a, b);
	}
	sort(e.begin(), e.end(), [&](const edge &x, const edge &y) {
		return x.a < y.a;
	});
	vector<node*> t(n + m);
	for (int i = 0; i < n; i++) {
		t[i] = new node(0, i);
	}
	int ans = 1e9;
	for (int i = 0; i < m; i++) {
		int a = e[i].a, b = e[i].b;
		int x = e[i].x, y = e[i].y;
		if (find_root(t[x]) != find_root(t[y])) {
			t[i + n] = new node(b, i + n);
			link(t[x], t[i + n]);
			link(t[y], t[i + n]);
		} else {
			split(t[x], t[y]);
			if (b < t[y]->maxb) {
				node* v = find(t[y], t[y]->maxb);
				int id = v->id - n;
				int vx = e[id].x, vy = e[id].y;
				cut(t[vx], v), cut(t[vy], v);
				t[i + n] = new node(b, i + n);
				link(t[x], t[i + n]);
				link(t[y], t[i + n]);
			}
		}
		if (find_root(t[0]) == find_root(t[n - 1])) {
			split(t[0], t[n - 1]);
			ans = min(ans, a + t[n - 1]->maxb);
		}
	}
	if (ans == 1e9) {
		cout << "-1" << endl;
	} else {
		cout << ans << endl;
	}
	return 0;
} 
posted @ 2019-09-26 19:27  shellpicker  阅读(195)  评论(0编辑  收藏  举报