【AMPPZ2014】【BZOJ4144】Petrol

Description

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

第一行包含三个正整数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，表示一个询问。
Output

输出q行。第i行输出第i个询问的答案，如果可行，则输出TAK，否则输出NIE。
Sample Input

6 4 5

1 5 2 6

1 3 1

2 3 2

3 4 3

4 5 5

6 4 5

4

1 2 4

2 6 9

1 5 9

6 5 8

Sample Output

TAK

TAK

TAK

NIE
HINT

Source

鸣谢Claris上传

先对每个点求出最近的加油站到这里的距离
然后我们可以把原图的边权替换成dis[u]+dis[v]+w之后排序
然后把询问按边权排序
允许的边就加入,并查集维护连通性

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#define MAXN 200010
#define GET (ch>='0'&&ch<='9')
#define LL long long
#define Qsize 1048575
using namespace std;
const LL MAXINT=1ll<<60;
int n,s,m,top,Q,size;
int f[MAXN];
int c[MAXN],vis[MAXN];
LL dis[MAXN];
int head=1,tail,q[Qsize+1];
int find(int x) {return f[x]==x?f[x]:f[x]=find(f[x]);}
void Union(int a,int b)
{
    int x=find(a),y=find(b);
    if (x!=y)   f[x]=y;
}
struct edge
{
    int to,w;
    edge *next;
}e[MAXN<<1],*prev[MAXN];
struct Edge
{
    int u,v,w;
    bool operator <(const Edge& a)const {return w<a.w;}
}E[MAXN];
struct Query
{
    int u,v,w,id,ans;
    bool operator <(const Query& a)const    {return w<a.w;}
}que[MAXN];
bool cmp(Query a,Query b)   {return a.id<b.id;}
void insert(int u,int v,int w)  {e[++top].to=v;e[top].next=prev[u];prev[u]=&e[top];e[top].w=w;}
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
int main()
{
    in(n);in(s);in(m);
    for (int i=1;i<=n;i++)  f[i]=i,dis[i]=MAXINT;
    for (int i=1;i<=s;i++)  in(c[i]),dis[c[i]]=0,q[++tail]=c[i],vis[c[i]]=1,size++;
    for (int i=1;i<=m;i++)  in(E[i].u),in(E[i].v),in(E[i].w),insert(E[i].u,E[i].v,E[i].w),insert(E[i].v,E[i].u,E[i].w);
    while (size)
    {
        int x=q[head++];head&=Qsize;size--;
        for (edge *i=prev[x];i;i=i->next)
            if (dis[x]+i->w<dis[i->to])
            {
                dis[i->to]=dis[x]+i->w;
                if (!vis[i->to])
                {
                    vis[i->to]=1;size++;
                    if (dis[i->to]<dis[q[head]])    q[head=(head+Qsize)&Qsize]=i->to;
                    else    q[tail=(tail+1)&Qsize]=i->to;
                }
            }
        vis[x]=0;
    }
    for (int i=1;i<=m;i++)
        if (dis[E[i].u]+dis[E[i].v]<MAXINT) E[i].w+=dis[E[i].u]+dis[E[i].v];
        else    E[i].w=0x7fffffff;
    sort(E+1,E+m+1);in(Q);int num=1;
    for (int i=1;i<=Q;i++)  in(que[i].u),in(que[i].v),in(que[i].w),que[i].id=i;sort(que+1,que+Q+1);
    for (int i=1;i<=Q;i++)
    {
        while (num<=m&&E[num].w<=que[i].w)  Union(E[num].u,E[num].v),num++;
        que[i].ans=(find(que[i].u)==find(que[i].v));
    }
    sort(que+1,que+Q+1,cmp);
    for (int i=1;i<=Q;i++)  puts(que[i].ans?"TAK":"NIE");
}