HDU 5361 In Touch (优先队列+并查集优化)



题意：求源点到所有点的最短距离。每个点只能到达与它相距l到r之间的点，且这个点到达这些点的花费都为c。

思路：这道题和普通最短路的区别是这个图有可能是个极为稠密的图，那么对于dijkstra算法来说时间复杂度太高，

注意到这题有一点特殊之处，每一点连出去的边的边权是相同的，也就是说一个点与源点之间的最短路只会更新一次，也就是说之后我们就可以不考虑更新这个点的最短路了，所以我们可以考虑用dijkstra算法的思想，用优先队列维护当前dist+cost最小的点，然后更新与它连边的那些点的最短路，那么我们剩下的问题就是考虑如何优化使得不再访问已更新的点。

我用的方法是并查集，如果一个点被更新过，那么就把它与它之后的最近的一个未更新的点进行merge操作，那么我们就做到了不再访问已更新过的点。

ps：一开始数组开小了没显示re竟然是tle，害我优化了一个小时结果一看是数组开小了，泪奔┭┮﹏┭┮

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<map>
#include<set>
#include<ctime>
#define eps 1e-6
#define LL long long
#define pii pair<int, int>
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;

const int maxn = 200000 + 1000;
const LL INF = 1e15;
int n, fa[maxn], L[maxn], R[maxn];
struct Node {
	LL d, c;
	int id;
	bool operator < (const Node& A) const {
		return (d+c) > (A.d+A.c);
	}
} node[maxn];
priority_queue<Node> q;
int Find(int x) {
	if(x==fa[x]) return x;
	return fa[x] = Find(fa[x]);
}
void merge(int x, int y) {
	int fa_x = Find(x), fa_y = Find(y);
	if(fa_x != fa_y) fa[fa_x] = fa_y;
}
void init() {
	for(int i = 1; i <= n+1; i++) fa[i] = i;
	for(int i = 1; i <= n; i++) node[i].d = -1, node[i].id = i;
	node[1].d = 0;
	fa[1] = 2;
	while(!q.empty()) q.pop();
}
void update(Node x, int dir) {
	int l = x.id + dir*L[x.id], r = x.id + dir*R[x.id];
	if(l > r) swap(l, r);
	l = max(1, l), r = min(r, n);
	if(l>n || r<=1) return;
	while(l<=r) {
		l = Find(l);
		if(l > r) break;
		node[l].d = x.d + x.c;
		q.push(node[l]);
		merge(l, l+1);
		l++;
	}
}
int main() {
    //freopen("input.txt", "r", stdin);
	int T; cin >> T;
	while(T--) {
		cin >> n;
		init();
		for(int i = 1; i <= n; i++) scanf("%d", &L[i]);
		for(int i = 1; i <= n; i++) scanf("%d", &R[i]);
		for(int i = 1; i <= n; i++) scanf("%I64d", &node[i].c);
		q.push(node[1]);
		while(!q.empty()) {
			Node tmp = q.top(); q.pop();
			update(tmp, -1);
			update(tmp, 1);
		}
		putchar('0');
		for(int i = 2; i <= n; i++) printf(" %I64d", node[i].d);
        puts("");
	}
    return 0;
}