Roads and Planes 拓扑排序+最短路

https://ac.nowcoder.com/acm/contest/1055/D

思路：
因为题目明确陆地是正边权，航空有负权边，所以可以把一个个陆地连通块看出一个点，然后用拓扑排序求最短路。

#include <stdio.h>
#include <queue>
#include <string.h>
#include <vector>
using namespace std;
int head[25001],nex[150010],ver[150010],d[150010];
int dis[25001],in[25001],belong[25001],tot,t,r,p,s,cnt;
bool vis[25001];
vector<int>team[25001];
int read()
{
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){s=s*10+ch-'0';ch=getchar();}
	return s*w;
}
void add(int u,int v,int z)
{
    tot++;
    nex[tot]=head[u];
    ver[tot]=v;
    d[tot]=z;
    head[u]=tot;
}
void dfs(int x)                    //查到连通块
{
    belong[x]=cnt;                //给连通块标记
    team[cnt].push_back(x);       //将属于自己一块的加入，这样可以求每个连通块内的最短路
    for(int i=head[x];i;i=nex[i])
    {
        int v=ver[i];
        if(!belong[v])dfs(v);
    }
}
void dijkstra()
{
    queue<int>q;                                    //用来存入度为0的点
    priority_queue<pair<int,int> >que;              //用来计算连通块最短路
    memset(dis,0x7f,sizeof(dis));
    q.push(belong[s]);                              //将起点先加入
    for(int i=1;i<=cnt;i++)if(!in[i])q.push(i);
    dis[s]=0;
    while(!q.empty())
    {
        int u=q.front();q.pop();
        for(int i=0;i<team[u].size();i++)           //对于入度为0的点，将它的连通块所有点输入
            que.push(make_pair(-dis[team[u][i]],team[u][i]));
        while(!que.empty())                      //求最短路
        {
            int u=que.top().second;que.pop();
            if(vis[u])continue;
            vis[u]=1;
            for(int i=head[u];i;i=nex[i])
            {
                int v=ver[i];
                if(dis[v]>dis[u]+d[i])
                {
                    dis[v]=dis[u]+d[i];
                    if(belong[u]==belong[v])que.push(make_pair(-dis[v],v));   //两个点属于同一个连通块
                }
                if(belong[u]!=belong[v])   //属于不同连通块
                {
                    in[belong[v]]--;
                    if(!in[belong[v]])q.push(belong[v]);
                }
            }
        }
    }
}
int main()
{
    int u,v,z;
    scanf("%d%d%d%d",&t,&r,&p,&s);
    for(int i=1;i<=r;i++)
    {
        u=read(),v=read(),z=read();
        add(u,v,z),add(v,u,z);
    }
    for(int i=1;i<=t;i++)
    {
        if(!belong[i])
            cnt++,dfs(i);
    }
    for(int i=1;i<=p;i++)
    {
        u=read(),v=read(),z=read();
        add(u,v,z);
        in[belong[v]]++;
    }
    dijkstra();
    for(int i=1;i<=t;i++)
        if(dis[i]>0x3f3f3f3f)printf("NO PATH\n");
        else printf("%d\n",dis[i]);
}