bzoj4144 [AMPPZ2014]Petrol spfa+并查集

Description

---

给定一个n个点、m条边的带权无向图，其中有s个点是加油站。
每辆车都有一个油量上限b，即每次行走距离不能超过b，但在加油站可以补满。
q次询问，每次给出x,y,b，表示出发点是x，终点是y，油量上限为b，且保证x点和y点都是加油站，请回答能否从x走到y。

第一行包含三个正整数n,s,m(2<=s<=n<=200000,1<=m<=200000)，表示点数、加油站数和边数。
第二行包含s个互不相同的正整数c[1],c[2],…cs，表示每个加油站。
接下来m行，每行三个正整数u[i],v[i],di，表示u[i]和v[i]之间有一条长度为d[i]的双向边。
接下来一行包含一个正整数q(1<=q<=200000)，表示询问数。
接下来q行，每行包含三个正整数x[i],y[i],bi，表示一个询问

Solution

---

解法非常巧妙，不知道多源最短路可以这么玩

注意到我们只需要保留加油站即可。由常识可得，一段a到b的路径上所有点的最近加油站 一定为a或b。也即是我们从a出发，沿着以a为起点的一条最短路走到某个最短路上的点x，然后走一条边w到点y，再沿着y到b的最短路走到终点

我们把所有的加油站作为起点跑多源最短路，记from[x]为距离点x最近的加油站。我们对于一条原图的边(x,y,w)连(from[x],from[y],dis[x]+dis[y]+w)
考虑离线询问做，用并查集维护连通性

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))
#define fi first
#define se second

typedef std:: pair <int,int> pair;
const int N=400005;
const int E=800005;

struct edge {int x,y,w,next;} e[E],g[E];
struct Q {int x,y,w,id;} q[N];

int ans[N],fa[N],c[N],from[N],dis[N],n,s,m;
int ls[N],edCnt,gdCnt;

bool vis[N];

int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}

void add_edge(int x,int y,int w) {
    e[++edCnt]=(edge) {x,y,w,ls[x]}; ls[x]=edCnt;
    e[++edCnt]=(edge) {y,x,w,ls[y]}; ls[y]=edCnt;
}

void spfa() {
    std:: queue <int> que;
    fill(dis,63);
    rep(i,1,s) {
        que.push(c[i]); vis[c[i]]=1;
        dis[c[i]]=0; from[c[i]]=c[i];
    }
    for (;!que.empty();) {
        int now=que.front(); que.pop();
        for (int i=ls[now];i;i=e[i].next) {
            if (dis[now]+e[i].w<dis[e[i].y]) {
                dis[e[i].y]=dis[now]+e[i].w;
                from[e[i].y]=from[now];
                if (!vis[e[i].y]) {
                    vis[e[i].y]=1;
                    que.push(e[i].y);
                }
            }
        } vis[now]=0;
    }
}

int find(int x) {
    if (!fa[x]) return x;
    return fa[x]=find(fa[x]);
}

bool merge(int x,int y) {
    int fx=find(x),fy=find(y);
    if (fx==fy) return false;
    fa[fx]=fy;
    return true;
}

bool cmp1(edge a,edge b) {
    return a.w<b.w;
}

bool cmp2(Q a,Q b) {
    return a.w<b.w;
}

int main(void) {
    n=read(); s=read(); m=read();
    rep(i,1,s) c[i]=read();
    rep(i,1,m) {
        int x=read(),y=read(),w=read();
        add_edge(x,y,w);
    }
    spfa();
    for (int i=1;i<=edCnt;i+=2) {
        if (from[e[i].x]!=from[e[i].y]) {
            g[++gdCnt]=(edge) {from[e[i].x],from[e[i].y],e[i].w+dis[e[i].x]+dis[e[i].y]};
        }
    }
    std:: sort(g+1,g+gdCnt+1,cmp1);
    int T=read();
    rep(i,1,T) q[i]=(Q) {read(),read(),read(),i};
    std:: sort(q+1,q+T+1,cmp2); int now=1;
    rep(i,1,T) {
        while (g[now].w<=q[i].w&&now<=gdCnt) {
            merge(g[now].x,g[now].y);
            now++;
        }
        if (find(q[i].x)==find(q[i].y)) ans[q[i].id]=1;
        else ans[q[i].id]=0;
    }
    rep(i,1,T) ans[i]?puts("TAK"):puts("NIE");
    return 0;
}