bzoj4144 [AMPPZ2014]Petrol（spfa+贪心+离线+并查集）

首先对于每个点i，记离它最近的加油站为from[i],我们可以贪心的发现，在加油站a到加油站b的最短路（只经过非加油站）上，所有的点的from一定是这样的：前一部分都是a，后一部分都是b。也就是说只存在这样的一条边，他的两端点的from不同，记这条边的边权为w，则a到b的最短路为dis[a]+w+dis[b]。（dis[i]表示i到离它最近的加油站的距离）于是我们以所有加油站为源点，做一遍多源最短路，求出所有的from和dis，然后枚举每一条边，如果两端点x,y的from不同，则dis[x]+dis[y]+w可能为from[x]到from[y]的最短路，加入这条新边。然后把询问离线，一边做最小生成树一边回答询问就好啦。
如果强制在线，也可以做完最小生成树来树上倍增求路径最大值。（貌似建出kruskal重构树，lca就是最大值也可以哟）

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,s,m,h[N],num=1,dis[N],from[N],pe[N],tot=0,fa[N];
bool inq[N],ans[N];
struct edge{
    int fr,to,nxt,val;
}data[N<<1];
struct Edge{
    int x,y,val,id;
}e[N],qu[N];
inline void add(int x,int y,int val){
    e[++tot].x=x;e[tot].y=y;e[tot].val=val;
}
inline bool cmp(Edge x,Edge y){return x.val<y.val;}
inline void spfa(){
    queue<int>q;memset(dis,inf,sizeof(dis));
    for(int i=1;i<=s;++i) q.push(pe[i]),dis[pe[i]]=0,inq[pe[i]]=1,from[pe[i]]=pe[i];
    while(!q.empty()){
        int x=q.front();q.pop();inq[x]=0;
        for(int i=h[x];i;i=data[i].nxt){
            int y=data[i].to;
            if(dis[x]+data[i].val<dis[y]){
                dis[y]=dis[x]+data[i].val;from[y]=from[x];
                if(!inq[y]) q.push(y),inq[y]=1;
            }
        }
    }
}
inline int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
int main(){
//  freopen("a.in","r",stdin);
    n=read();s=read();m=read();
    for(int i=1;i<=s;++i) pe[i]=read();
    while(m--){
        int x=read(),y=read(),val=read();
        data[++num].to=y;data[num].fr=x;data[num].nxt=h[x];h[x]=num;data[num].val=val;
        data[++num].to=x;data[num].fr=y;data[num].nxt=h[y];h[y]=num;data[num].val=val;
    }spfa();
    for(int i=1;i<=num;i+=2){
        int x=data[i].fr,y=data[i].to;if(from[x]==from[y]) continue;
        add(from[x],from[y],dis[x]+dis[y]+data[i].val);
    }sort(e+1,e+tot+1,cmp);m=read();
    for(int i=1;i<=m;++i) qu[i].x=read(),qu[i].y=read(),qu[i].val=read(),qu[i].id=i;
    sort(qu+1,qu+m+1,cmp);int p=1;for(int i=1;i<=n;++i) fa[i]=i;
    for(int i=1;i<=m;++i){
        while(p<=tot&&e[p].val<=qu[i].val){
            int xx=find(e[p].x),yy=find(e[p].y);
            if(xx!=yy) fa[xx]=yy;++p;
        }ans[qu[i].id]= find(qu[i].x)==find(qu[i].y);
    }for(int i=1;i<=m;++i) puts(ans[i]?"TAK":"NIE");
    return 0;
}