[蓝桥杯练习]蓝桥王国

单源最短路径问题-dj

#include<bits/stdc++.h>
#define ll long long  
using namespace std;
const int N=3e5+5,M=1e6+5;
const ll INF=0x7f7f7f7f7f7f7f;//7个7f没问题,INF <= INF+x 
struct edge{
	int to;ll w;
	edge(int end,ll cost){to=end;w=cost;}
};
struct node{
	int id;ll setdis;//id：结点；setdis：这个结点到集合内的点最短的距离
	node(int num,ll len){id=num;setdis=len;}
	bool operator <(const node& cur)const{return setdis > cur.setdis;}//小根堆按照dis的升序来排序自定义的node对象
    //{return cur.dis < dis;}  //大根堆按照dis的降序来排序自定义的node对象
    //要将 < 运算符重载为适用于小根堆需要确保当 新结点cur的dis值 < 当前对象的dis值时，函数返回true。
    //*this（隐式参数）代表当前对象，即调用运算符重载函数的对象
    //将具有较小dis值的节点被放置在更接近堆顶的层次
};
vector<vector<edge>>adjtable(M);
priority_queue<node>wait;//小根堆,优先队列，存结点信息,弹出距离集合最近节点
int n,m;
ll setdis[N];//记录所有结点到集合的最近距离
bool hasmin[M];//hasmin[i]=true表示到结点i的最短路径已经找到
bool haspath[N];//如果不想用INF判定,则开个数组表示是否有通路
int pre[N];//记录前驱结点，用于生成路径
void print_path(int start,int end){		  //打印从s到t的最短路
    if(start==end){cout<<start;return;}  	   //打印起点
    print_path(start,pre[end]);                //先打印前一个点
    cout<<end;			                      //后打印当前点。最后打印的是终点t
}
void dj(){
	while(!wait.empty()){
		node cur = wait.top();wait.pop();//小根堆,弹出距起点start距离最小的结点minnode
		if(hasmin[cur.id])continue;//丢弃已经找到最短路径的结点
		hasmin[cur.id]=true;haspath[cur.id]=true;//标记已访问 和有最短路径 
		for(edge adj:adjtable[cur.id]){//检查当前点的所有邻边
			if(hasmin[adj.to])continue;//丢弃已经找到最短路径的邻居结点 
			if(setdis[adj.to] > adj.w + cur.setdis){//如果邻点到起点的距离 大于 若当前点到起点的距离 + 邻点到当前点的边权 ,则更新邻点到起点的距离为中转当前点再过去的两端距离和 
				setdis[adj.to] = adj.w + cur.setdis;//邻点最短距离=邻边权重+最近点到起点的距离
				wait.push(node(adj.to,setdis[adj.to]));//扩展新的邻居，放到优先队列中
				//pre[adj.to]=cur.id;		//如果有需要，记录路径
}	}	}	
// print_path(s，n);          			  //如果有需要，打印路径： 起点1，终点n
}
int main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	memset(setdis,INF,sizeof(setdis));//memset是按字节赋值,所以应当是0x7f 
//	for(ll x:sdis)x=0x7f7f7f7f7f7f7f;  //循环应该是赋真值 
	memset(hasmin,false,sizeof(hasmin));
	cin>>n>>m;
	while(m--){
		int v1,v2;ll w;cin>>v1>>v2>>w;
		adjtable[v1].push_back(edge(v2,w));
	// adjtable[v2].push_back(edge(v1,w));    //无向图或重边
	}
	int start_id;//cin>>start_id;
	start_id=1;
	//haspath[start_id]可以不加,但是最好加上 
	wait.push(node(start_id,0));//起点进队列,起点到自己的距离是0
	setdis[start_id]=0;
	dj();
	for(int i=1;i<=n;++i){//输出到每个点的最短路径
//		if(setdis[i]>=0x7f7f7f7f7f7f7f)cout<<"-1 ";
		if(!haspath[i])cout<<"-1 ";
		else cout<<setdis[i]<<" ";
	}
	return 0;
}