【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之后排序
然后把询问按边权排序
允许的边就加入,并查集维护连通性
#includeint u,v,w,id,ans;
bool operator <(const Query& a)const {return wreturn a.idint 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->wto])
{
dis[i->to]=dis[x]+i->w;
if (!vis[i->to])
{
vis[i->to]=1;size++;
if (dis[i->to]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]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");
}