道路与航线

一道类似缩点的好题,先按道路缩点 然后将缩点以后的图按照航线做DAG

在DAG上先跑topsort 在每一个团内部跑dijkstra,同时更新top点 很有意思的一道题目

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 3e5+10;
const int inf = 0x3f3f3f3f;
const int mod = 1e9+7;

int n,m1,m2,st;

int e[N],ne[N],w[N],h[N],idx;
void add(int a,int b,int c){
	e[idx] = b,ne[idx] = h[a],w[idx] = c,h[a] = idx++;
}


int din[N];
int id[N];
vector<int>block[N];
bool df[N];
int cnt;
int dist[N];
bool vis[N];

void dfs(int u,int ids){
	df[u] = true;
	id[u] = ids;
	block[ids].push_back(u);
	
	for(int i=h[u];~i;i=ne[i]){
		int j = e[i];
		if(!df[j])dfs(j,ids);
	} 
}
queue<int>q;
void dijkstra(int ids)
{
	
	priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>>heap;
	for(auto &t:block[ids])heap.push({dist[t],t});
	
	
	while(heap.size()){
		
		auto t = heap.top();heap.pop();
		
		int ver = t.second,distance = t.first;
		
		if(vis[ver])continue;
		vis[ver] = true;
		
		for(int i=h[ver];~i;i=ne[i]){
			int j = e[i];
			//cout<<ver<<" "<<" "<<j<<"\n";
			
			if(dist[j]>distance+w[i]){
				dist[j] = dist[ver]+w[i];
				if(id[j]==id[ver]){
					heap.push({dist[j],j});
				}
			}
			if(id[j]!=id[ver]&&(--din[id[j]]==0))q.push(id[j]);
		}
		
		
	}
	
}

void topsort()
{
	
	for(int i=1;i<=cnt;++i)
	 if(!din[i]){q.push(i);}
	memset(dist,0x3f,sizeof dist);
	memset(vis,0,sizeof vis); 
	dist[st] = 0;
	while(q.size()){
		
		auto t = q.front();
		q.pop();
		//cout<<t<<"??\n";
		dijkstra(t);
		
	}
	
	
}

void solve()
{
	cin>>n>>m1>>m2>>st;
	memset(h,-1,sizeof h);
	for(int i=1;i<=m1;i++){
		int a,b,c;cin>>a>>b>>c;
		add(a,b,c),add(b,a,c);
	}
	
	//缩点
	
	for(int i=1;i<=n;i++)if(!df[i])dfs(i,++cnt);

	
	for(int i=1;i<=m2;i++){
		int a,b,c;cin>>a>>b>>c;
		add(a,b,c);
		din[id[b]]++;
	}
	
	topsort();
	
	
	for(int i=1;i<=n;i++)
	 if(dist[i]>0x3f3f3f3f/2)cout<<"NO PATH\n";
	 else cout<<dist[i]<<"\n";
	
	

}

signed main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int _;
	//cin>>_;
	_ = 1;
	while(_--)solve();
	return 0;
}
相关推荐
qq_4298796727 分钟前
省略号和可变参数模板
开发语言·c++·算法
优秀的颜1 小时前
计算机基础知识(第五篇)
java·开发语言·分布式
CodeWithMe1 小时前
【C/C++】std::vector成员函数清单
开发语言·c++
uyeonashi1 小时前
【QT控件】输入类控件详解
开发语言·c++·qt
飞川撸码1 小时前
【LeetCode 热题100】网格路径类 DP 系列题:不同路径 & 最小路径和(力扣62 / 64 )(Go语言版)
算法·leetcode·golang·动态规划
Neil今天也要学习2 小时前
永磁同步电机参数辨识算法--IPMSM拓展卡尔曼滤波全参数辨识
单片机·嵌入式硬件·算法
yzx9910132 小时前
基于 Q-Learning 算法和 CNN 的强化学习实现方案
人工智能·算法·cnn
iCxhust3 小时前
Prj10--8088单板机C语言8259测试(1)
c语言·开发语言
亮亮爱刷题3 小时前
算法练习-回溯
算法
眼镜哥(with glasses)3 小时前
蓝桥杯 国赛2024python(b组)题目(1-3)
数据结构·算法·蓝桥杯