首先还是说这道题的收获吧
一定要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_queuepq; 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_queuepq; 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…
所以我们求出来各个加油站之间的最短路,然后做最小生成树…查询的时候倍增查询…
由于原图不联通,所以我们需要一个虚根…
代码如下:
#include#include #include #include #include //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) ='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 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] =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; }