2200: [Usaco2011 Jan]道路和航线 （拓扑排序+dijstra）

  1 #include<bits/stdc++.h>
  2 using namespace std;
  3 
  4 int t,r,p,s;
  5 typedef pair<int,int> pii;
  6 const int maxn = 5e4+5;
  7 vector<int>belong[maxn];
  8 struct Node
  9 {
 10     int x,y,val,next;
 11     Node(int x=0,int y=0,int val=0,int next=0):x(x),y(y),val(val),next(next) {}
 12 } node[2][maxn<<2];
 13 
 14 int head[2][maxn];
 15 int cnt[2];
 16 int tot;
 17 void add(int x,int y,int val,int t)
 18 {
 19     node[t][++cnt[t]].x = x;
 20     node[t][cnt[t]].y = y;
 21     node[t][cnt[t]].val = val;
 22     node[t][cnt[t]].next = head[t][x];
 23     head[t][x] = cnt[t];
 24 }
 25 int cn[maxn];
 26 int ind[maxn];
 27 int dist[maxn];
 28 bool vis[maxn];
 29 void dfs(int x)
 30 {
 31     cn[x] = tot;
 32     belong[tot].push_back(x);
 33     for(int i=head[0][x]; i; i=node[0][i].next)
 34     {
 35         int y = node[0][i].y;
 36         if(cn[y])
 37             continue;
 38         dfs(y);
 39     }
 40 }
 41 void solve(int s)
 42 {
 43     queue<int>que;
 44     while(!que.empty())
 45         que.pop();
 46     for(int i=1; i<=tot; i++)
 47         if(!ind[i])
 48             que.push(i);
 49 //    for(int i=1; i<=tot; i++)
 50 //    {
 51 //        if(!ind[i] && i != cn[s])
 52 //        {
 53 //            for(int j=0; j<belong[i].size(); j++)
 54 //            {
 55 //                for(int k=head[1][belong[i][j]]; k; k=node[1][k].next)
 56 //                {
 57 //                    --ind[cn[node[1][k].y]];
 58 //                }
 59 //            }
 60 //            ind[i] = -1;
 61 //            //printf("%d ------ %d\n",ind[i],i);
 62 //            i=0;
 63 //        }
 64 //    }
 65 //    que.push(cn[s]);
 66     memset(dist,0x3f,sizeof(dist));
 67     dist[s] = 0;
 68     priority_queue<pii,vector<pii>,greater<pii> >p;
 69     while(!que.empty())
 70     {
 71         int k = que.front();
 72         que.pop();
 73         while(!p.empty())
 74             p.pop();
 75         int len = belong[k].size();
 76         for(int i=0; i<len; i++)
 77         {
 78             if(dist[belong[k][i]] < 0x3f3f3f3f)
 79                 p.push(pii(dist[belong[k][i]],belong[k][i]));
 80         }
 81         while(!p.empty())
 82         {
 83             pii t = p.top();
 84             p.pop();
 85             if(vis[t.second])
 86                 continue;
 87             vis[t.second] = 1;
 88             for(int i=head[0][t.second]; i; i=node[0][i].next)
 89             {
 90                 if(dist[t.second] + node[0][i].val < dist[node[0][i].y])
 91                 {
 92                     dist[node[0][i].y] = dist[t.second] + node[0][i].val;
 93                     p.push(pii(dist[t.second] + node[0][i].val,node[0][i].y));
 94                 }
 95             }
 96             for(int i=head[1][t.second]; i; i=node[1][i].next)
 97             {
 98                 dist[node[1][i].y] = min(dist[node[1][i].y],dist[t.second]+node[1][i].val);
 99             }
100         }
101         for(int i=0; i<len; i++)
102         {
103             for(int j=head[1][belong[k][i]]; j; j=node[1][j].next)
104             {
105                 if(--ind[cn[node[1][j].y]] == 0)
106                     que.push(cn[node[1][j].y]);
107             }
108         }
109     }
110 }
111 
112 int main()
113 {
114     cnt[0] = cnt[1] = 0;
115     scanf("%d%d%d%d",&t,&r,&p,&s);
116     for(int i=1; i<=t; i++)
117         belong[i].clear();
118     for(int i=1; i<=r; i++)
119     {
120         int u,v,k;
121         scanf("%d%d%d",&u,&v,&k);
122         add(u,v,k,0);
123         add(v,u,k,0);
124     }
125 
126     for(int i=1; i<=p; i++)
127     {
128         int u,v,k;
129         scanf("%d%d%d",&u,&v,&k);
130         add(u,v,k,1);
131     }
132     tot = 0;
133     for(int i=1; i<=t; i++)
134     {
135         if(!cn[i])
136         {
137             tot++;
138             dfs(i);
139         }
140     }
141 
142 
143     for(int i=1; i<=cnt[1]; i++)
144     {
145         ind[cn[node[1][i].y]]++;
146     }
147     solve(s);
148     for(int i=1; i<=t; i++)
149     {
150         if(dist[i] == 0x3f3f3f3f)
151             printf("NO PATH\n");
152         else
153             printf("%d\n",dist[i]);
154     }
155 }