道路与航线 BZOJ2200

H. Delivery Route
算法竞赛进阶指南P358例题

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
const int MAX_N=50010;
const int INF=0x3f3f3f3f;
vector<int>v[MAX_N],w[MAX_N];
int fa[MAX_N];
queue<int>qr;
struct skt{
 int x,d;
 friend bool operator <(skt a,skt b){
  return a.d>b.d;
 }
};
int cnt,n;
int dis[MAX_N],vis[MAX_N];
int sum[MAX_N];
void dfs(int x,int st){
 v[st].push_back(x);
 fa[x]=st;
 //cout<<st<<" "<<x<<"!!\n";
 for(int i=0;i<v[x].size();i++){
  int y=v[x][i];
  if(fa[y])
  continue;
  dfs(y,st);
 }
}
void Dijkstra(int st){
 priority_queue<skt>q;  
 for(int i=0;i<v[st].size();i++){
  int y=v[st][i];
  skt k;
  k.d=dis[y];
  k.x=y;
  q.push(k);
 }
 while(!q.empty()){
  skt t=q.top();
  q.pop();
  int x=t.x;
  if(vis[x])
  continue;
  vis[x]=true;
  for(int i=0;i<v[x].size();i++){
   int  y=v[x][i];
   int cost=w[x][i];
   if(fa[x]==fa[y]){
    if(dis[x]+cost<dis[y]){
     dis[y]=dis[x]+cost;
     skt b;
     b.x=y;
     b.d=dis[y];
     q.push(b); 
    }
   }
   else{
    if(dis[x]!=INF&&dis[x]+cost<dis[y])
    dis[y]=dis[x]+cost;
    sum[fa[y]]--;
    if(sum[fa[y]]==0)
    qr.push(fa[y]);
   }
  }
 }
}
void topo(){
 for(int i=n+1;i<=cnt;i++){
  if(!sum[i])
  qr.push(i);
 }
 while(!qr.empty()){
  int x=qr.front();
  qr.pop();
  Dijkstra(x);
 }
}
int main(void){
 int m1,m2,x,y,z,s,i;
 memset(dis,INF,sizeof(dis));
 scanf("%d%d%d%d",&n,&m1,&m2,&s);
 dis[s]=0;
 for(i=0;i<m1;i++){
  scanf("%d%d%d",&x,&y,&z);
  v[x].push_back(y);
  w[x].push_back(z);
  v[y].push_back(x);
  w[y].push_back(z);
 }
 cnt=n;
 for(i=1;i<=n;i++){
  if(!fa[i]){
   //cout<<i<<"!\n";
   dfs(i,++cnt);
  }
 }
 for(i=0;i<m2;i++){
  scanf("%d%d%d",&x,&y,&z);
  v[x].push_back(y);
  w[x].push_back(z);
  sum[fa[y]]++;
 }
 topo();
 for(i=1;i<=n;i++){
  if(dis[i]==INF)
  printf("NO PATH\n");
  else
  printf("%d\n",dis[i]);
 }
 return 0;
}