1. 程式人生 > >51nod 1443 路徑和樹——最短路生成樹

51nod 1443 路徑和樹——最短路生成樹

con tar def etc break air getchar() str memset

題目:http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1443

不只是做一遍最短路。還要在可以選的邊裏選最短的才行。

以為是求一遍最短路,然後按邊權對邊排序,哪條邊兩邊的dis正好吻合,就把該邊的邊權加到ans裏,把兩端加到並查集裏。

但其實不對。因為忽略了方向。比如如果有多個點同樣地可以更新一個點,算的時候可能這多個點都因為那個點而被合到了並查集裏,但其實只能有一個被合進去。

其實只要松弛點的時候如果dis[v]==dis[k]+w[i],就把mn[v]對w[i]取min,最後把除了根的每個點的mn[ ]都算上就行了。

#include<iostream>
#include
<cstdio> #include<cstring> #include<algorithm> #include<queue> #define ll long long using namespace std; const int N=3e5+5; int n,m,hd[N],xnt,to[N<<1],nxt[N<<1],w[N<<1],st,mn[N]; ll dis[N],ans; bool vis[N]; priority_queue<pair<ll,int> > q; int rdn() {
int ret=0;bool fx=1;char ch=getchar(); while(ch>9||ch<0){if(ch==-)fx=0;ch=getchar();} while(ch>=0&&ch<=9) ret=(ret<<3)+(ret<<1)+ch-0,ch=getchar(); return fx?ret:-ret; } void add(int x,int y,int z) { to[++xnt]=y; nxt[xnt]=hd[x]; hd[x]=xnt; w[xnt]=z; to[
++xnt]=x; nxt[xnt]=hd[y]; hd[y]=xnt; w[xnt]=z; } void dj() { q.push(make_pair(0,st)); memset(dis,0x3f,sizeof dis); dis[st]=0; while(q.size()) { int k=q.top().second; q.pop(); while(vis[k]&&q.size())k=q.top().second,q.pop(); if(vis[k])break; vis[k]=1; for(int i=hd[k],v;i;i=nxt[i]) { if(dis[v=to[i]]>dis[k]+w[i]) { dis[v]=dis[k]+w[i]; q.push(make_pair(-dis[v],v)); mn[v]=w[i]; } else if(dis[v]==dis[k]+w[i]) mn[v]=min(mn[v],w[i]); } } } int main() { n=rdn(); m=rdn(); for(int i=1,u,v,z;i<=m;i++) { u=rdn(); v=rdn(); z=rdn(); add(u,v,z); } st=rdn(); memset(mn,0x3f,sizeof mn); dj(); for(int i=1;i<=n;i++) if(i!=st) ans+=mn[i]; printf("%lld\n",ans); return 0; }

51nod 1443 路徑和樹——最短路生成樹