BZOJ 4144: [AMPPZ2014]Petrol 最短路+最小生成树+倍增

时空隧道

首先还是说这道题的收获吧…

一定要dij一定要把点和dis都扔进堆里，之前都是开一个数组保存dis，这样会T：

TLE：

struct M{
int node;
friend bool operator < (M a,M b){
return dis[a.node]>dis[b.node];
}
M(int a=0){
node=a;
}
};
inline void dijkstra(void){
priority_queue<M>pq;
memset(vis,0,sizeof(vis));memset(dis,inf,sizeof(dis));
for(int i=1;i<=S;i++)
vis[c[i]]=1,dis[c[i]]=0,pq.push(M(c[i])),from[c[i]]=c[i];
while(!pq.empty()){
int top=pq.top().node;pq.pop();
for(int i=hd[top];i!=-1;i=nxt[i])
if(dis[to[i]]>dis[top]+w[i]){
dis[to[i]]=dis[top]+w[i],from[to[i]]=from[top];
if(!vis[to[i]])
vis[to[i]]=1,pq.push(to[i]);
}
}
}

快一点的：

struct M{
int node;
long long di;
friend bool operator < (M a,M b){
return a.di>b.di;
}
M(int a=0,long long b=0){
node=a;di=b;
}
};
inline void dijkstra(void){
priority_queue<M> pq;
for(int i=1;i<=S;i++)
dis[c[i]]=0,pq.push(M(c[i],0)),from[c[i]]=c[i];
while(!pq.empty()){
int top=pq.top().node;long long di=pq.top().di;pq.pop();
if(di!=dis[top])
continue;
for(int i=hd[top];i!=-1;i=nxt[i])
if(dis[to[i]]>dis[top]+w[i])
dis[to[i]]=dis[top]+w[i],
from[to[i]]=from[top],
pq.push(M(to[i],dis[to[i]]));
}
}

然后就是如果数组是long long其他相关数组也要开成long long…

一定要掌握好inf的大小…

不要随便用clock()…bzoj是会RE的…(调试的时候为了看时间用了clock，教的时候忘记删RE到死QAQ)…

分析：

我们考虑从一个点走到另一点显然是经过了若干个加油站的…那么这条路上的限制只和加油站之间的距离有关系…其他点都是打酱油的QAQ…

所以我们求出来各个加油站之间的最短路，然后做最小生成树…查询的时候倍增查询…

由于原图不联通，所以我们需要一个虚根…

代码如下：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
//by NeighThorn
#define inf 1000000000000000000
using namespace std;
const int maxn=200000+5,maxm=200000+5;
int n,S,m,tot,c[maxn],hd[maxn],to[maxm*4],nxt[maxm*4],cnt,q,from[maxn],fa[maxn][30],dep[maxn];
long long diss[maxn][30],dis[maxn],w[maxm*4];
struct M{
int node;
long long di;
friend bool operator < (M a,M b){
return a.di>b.di;
}
M(int a=0,long long b=0){
node=a;di=b;
}
};
struct edge{
int x,y;
long long ww;
friend bool operator < (edge a,edge b){
if(a.ww==b.ww)
return min(a.x,a.y)<min(b.y,b.x);
return a.ww<b.ww;
}
}lala[maxm*4],e[maxm];
inline int read(void){
char ch=getchar();int x=0;
while(!(ch>='0'&&ch<='9'))
ch=getchar();
while(ch>='0'&&ch<='9')
x=x*10+ch-'0',ch=getchar();
return x;
}
inline void add(int x,int y,long long s){
w[cnt]=s;to[cnt]=y;nxt[cnt]=hd[x];hd[x]=cnt++;
w[cnt]=s;to[cnt]=x;nxt[cnt]=hd[y];hd[y]=cnt++;
}
inline int find(int x){
return fa[x][0]==x?x:fa[x][0]=find(fa[x][0]);
}
inline void dijkstra(void){
priority_queue<M> pq;
for(int i=1;i<=S;i++)
dis[c[i]]=0,pq.push(M(c[i],0)),from[c[i]]=c[i];
while(!pq.empty()){
int top=pq.top().node;long long di=pq.top().di;pq.pop();
if(di!=dis[top])
continue;
for(int i=hd[top];i!=-1;i=nxt[i])
if(dis[to[i]]>dis[top]+w[i])
dis[to[i]]=dis[top]+w[i],from[to[i]]=from[top],pq.push(M(to[i],dis[to[i]]));
}
}
inline void kruskal(void){
sort(lala+1,lala+tot+1);
for(int i=1;i<=tot;i++){
int fx=find(lala[i].x),fy=find(lala[i].y);
if(fx!=fy)
fa[fx][0]=fy,add(lala[i].x,lala[i].y,lala[i].ww);
}
}
inline void dfs(int root,int f){
for(int j=1;j<=25;j++)
fa[root][j]=fa[fa[root][j-1]][j-1],diss[root][j]=max(diss[root][j-1],diss[fa[root][j-1]][j-1]);
for(int i=hd[root];i!=-1;i=nxt[i])
if(to[i]!=f)
fa[to[i]][0]=root,diss[to[i]][0]=w[i],dep[to[i]]=dep[root]+1,dfs(to[i],root);
}
inline bool redouble(int x,int y,int b){
long long ans=0;
if(dep[x]<dep[y])
swap(x,y);
int d=dep[x]-dep[y];
for(int i=25;i>=0;i--)
if(dep[fa[x][i]]>=dep[y]){
ans=max(ans,diss[x][i]),x=fa[x][i];
if(ans>b)
return false;
}
if(x==y)
return ans<=b;
for(int i=25;i>=0;i--)
if(fa[x][i]!=fa[y][i]){
ans=max(ans,max(diss[x][i],diss[y][i])),x=fa[x][i],y=fa[y][i];
if(ans>b)
return false;
}
ans=max(ans,max(diss[x][0],diss[y][0]));
return ans<=b;
}
signed main(void){
n=read();S=read();m=read();
memset(hd,-1,sizeof(hd));
for(int i=1,x;i<=S;i++)
c[i]=read();
for(int i=1,s,x,y;i<=m;i++)
x=read(),y=read(),s=read(),add(x,y,(long long)s),e[i].x=x,e[i].y=y,e[i].ww=s;
for(int i=1;i<=n;i++)
fa[i][0]=i,dis[i]=inf;
dijkstra();cnt=0;tot=0;
for(int i=1;i<=m;i++)
if(from[e[i].x]!=from[e[i].y])
lala[++tot].x=from[e[i].x],lala[tot].y=from[e[i].y],lala[tot].ww=dis[e[i].x]+dis[e[i].y]+e[i].ww;
memset(hd,-1,sizeof(hd));
for(int i=1;i<=S;i++)
lala[++tot].x=0,lala[tot].y=c[i],lala[tot].ww=inf;
kruskal();dep[0]=0;fa[0][0]=0;dfs(0,-1);
q=read();int x,y,b;
while(q--){
x=read();y=read();b=read();
if(!redouble(x,y,b))
puts("NIE");
else
puts("TAK");
}
return 0;
}

by >_< NeighThorn