hdu5361(最短路+并查集)

链接：点击打开链接

题意：有n个点排成一排，编号从1到n，每个点可以到自己[l, r]距离的位置，花费c，问一号店到其他点的最小花费

代码：

#pragma comment(linker, "/STACK:102400000,102400000")
#include <set>
#include <queue>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
const long long INF=0x3f3f3f3f3f3f3f3f;
typedef pair<long long,int> P;
int V;
int l[500005],r[500005],par[500005];
long long d[500005],w[500005];
void init(int n){
    int i;
    for(i=0;i<=2*n;i++){
        par[i]=i;
        d[i]=INF;
    }
}
int find(int x){
    if(par[x]==x)
    return par[x];
    return par[x]=find(par[x]);
}
void unite(int x,int y){
    x=find(x);
    y=find(y);
    if(x==y)
    return;
    par[x]=y;
}
struct edge{
    int to,cost;
};
void dijkstra(int s){
    int i,j,v,le,re;
    priority_queue<P,vector<P>,greater<P> >que;
    d[s]=w[s];                                  //应该再加一个边权，如果用点的距离
    que.push(P(d[s],s));                        //则不知道应该再往哪走
    while(!que.empty()){
        P p=que.top(); que.pop();
        v=p.second;
        for(i=-1;i<=1;i+=2){
            le=v+l[v]*i;
            re=v+r[v]*i;
            if(le>re)
            swap(le,re);
            le=max(1,le);
            re=min(V,re);
            for(j=le;j<=re;j++){
                j=find(j);                      //用并查集找出当前点上一个区间最后一个点
                if(j>re)
                break;
                if(d[j]>d[v]+w[j]){
                    d[j]=d[v]+w[j];
                    que.push(P(d[j],j));
                }
                unite(j,j+1);
            }
        }
    }
}
int main(){                                     //因为每个点所连的边的权值都相等，因此每经过
    int i,j,t;                                  //一个点一定不会在经过第二次，所以问题就变为
    scanf("%d",&t);                             //如何在最短路时直接找到没有更新过的点
    while(t--){
        scanf("%d",&V);
        for(i=1;i<=V;i++)
        scanf("%d",&l[i]);
        for(i=1;i<=V;i++)
        scanf("%d",&r[i]);
        for(i=1;i<=V;i++)
        scanf("%I64d",&w[i]);
        init(V);
        dijkstra(1);
        for(i=1;i<V;i++){
            if(d[i]==INF)
            printf("-1 ");
            else
            printf("%I64d ",d[i]-w[i]);
        }
        if(d[V]==INF)
        puts("-1");
        else
        printf("%I64d\n",d[V]-w[V]);
    }
    return 0;
}