题意:求源点到所有点的最短距离。每个点只能到达与它相距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;
}