BZOJ 4144 [AMPPZ2014]Petrol 最短路+最小產生樹__ac之路

首先可以很顯然的知道題目中只與加油站有關，也就是說其他的點都是無用的，所以先考慮如何取代其他無用的點，如下圖：

黑色代表加油站，紅色代表無用的點，現在要去掉紅色4號，那麼考慮對於1,2,3節點新的邊應該是什麼樣子。

考慮從3號節點到達1號，如果直接沿簡單路徑走，油量最小為7，而如果從3到2再到1，就只需要5就可以了，原因：

b<c&&b<a   所以a+b<a+c && b+c<a+c

顧這樣的走法是最優的，這樣一來我們就能得到一個性質，從當前節點到距離他最近的加油站再去往其他加油站不會更差。所以做多源最短路，記錄每一個點最近的加油站，然後如果一條邊連結的兩個點的最近的加油站不同，就建立一條串連著兩個加油站的邊，權值顯然

最後只需要做一次最小產生樹判聯性就好了。

#include<cstdio>#include<cstring>#include<iostream>#include<queue>#include<algorithm>#define maxn 200021#define inf 0x3fffffffusing namespace std;typedef pair<int,int>pii;int n,m,s,C[maxn],ans[maxn],Q,head[maxn],cnt;int dis[maxn],tot=1,f[maxn],vis[maxn],near[maxn];struct que{int x,y,d,id;bool operator<(const que& b)const{return d<b.d;}}qu[maxn];struct data{int a,b,c;bool operator<(const data& y)const{return c<y.c;}}nod[maxn*4];struct edge{int v,next,w;}e[maxn*4];void adde(int a,int b,int c){e[tot].v=b,e[tot].next=head[a],e[tot].w=c;head[a]=tot++;}int find(int x){return f[x]==x ? x : f[x]=find(f[x]);}void dijkstra(){priority_queue<pii,vector<pii>,greater<pii> >q;for(int i=1;i<=n;i++)dis[i]=inf;for(int i=1;i<=s;i++){dis[C[i]]=0;near[C[i]]=C[i];q.push(make_pair(0,C[i]));}while(!q.empty()){int u=q.top().second;q.pop();if(vis[u])continue;vis[u]=1;for(int v,i=head[u];i;i=e[i].next){if(dis[v=e[i].v]>dis[u]+e[i].w){dis[v]=dis[u]+e[i].w;near[v]=near[u];q.push(make_pair(dis[v],v));}}}}void build(){for(int i=1;i<=n;i++){for(int v,j=head[i];j;j=e[j].next){if(near[i]!=near[v=e[j].v]){nod[++cnt]=(data){near[i],near[v],dis[i]+dis[v]+e[j].w};}}}sort(nod+1,nod+1+cnt);}void solve(){int now=1;for(int i=1;i<=n;i++)f[i]=i;for(int i=1;i<=Q;i++){while(now<=cnt&&nod[now].c<=qu[i].d){int dx=find(nod[now].a),dy=find(nod[now].b);if(dx!=dy)f[dx]=dy;now++;}int dx=find(qu[i].x),dy=find(qu[i].y);if(dx==dy)ans[qu[i].id]=1;else ans[qu[i].id]=0;}for(int i=1;i<=Q;i++)if(ans[i])puts("TAK");else puts("NIE");}int main(){scanf("%d%d%d",&n,&s,&m);for(int i=1;i<=s;i++)scanf("%d",C+i);for(int a,b,c,i=1;i<=m;i++){scanf("%d%d%d",&a,&b,&c);adde(a,b,c);adde(b,a,c);}dijkstra();build();scanf("%d",&Q);for(int i=1;i<=Q;i++)scanf("%d%d%d",&qu[i].x,&qu[i].y,&qu[i].d),qu[i].id=i;sort(qu+1,qu+1+Q);solve();return 0;}