洛谷 P4779 P3371 单源最短路径 dijkstra堆优化(n+mlogm)

近期洛谷将P3371的单源最短路模板更新了,原因是管理员fstqwq的两位朋友在参加2018NOI时,由于用了SPFA去做单源最短路这部分的程序时,本来的100分变成了60,Ag->Cu.于是管理员决定将P3371升级,SPFA就过不去了.(所以我们以后还是用dijkstra的堆优化吧QWQ)
###题目: P4779 【模板】单源最短路径(标准版)
###题目: P3371 【模板】单源最短路径(弱化版)
这是以前的一个直接用小根堆写的程序,放到4779上去,错了.于是再写了一个用大根堆的程序.见下面那个

第一种版本:

//小根堆
#include
#include
#include
#include
#include
#include
#include
#define pa pair
using namespace std;
struct arr{
	int nd,nx,co;
}bot[1000000];
int head[11000];
int dis[11000];
int n,m,s,cnt;
inline void add(int u,int v,int w){bot[++cnt].nd=v;bot[cnt].co=w;bot[cnt].nx=head[u];head[u]=cnt;}
inline void dijkstra(int s){
	priority_queue<pa,vector<pa>,greater<pa> >q;
	//定义小根堆
	for(int i=1;i<=n;i++) dis[i]=2147483647;
	dis[s]=0;
	q.push(make_pair(0,s));
	while(!q.empty()){
		int now=q.top().second;
		q.pop();
		for(int i=head[now];i;i=bot[i].nx)
		 if(dis[now]+bot[i].co<dis[bot[i].nd]){
		 	dis[bot[i].nd]=dis[now]+bot[i].co;
		 	q.push(make_pair(dis[bot[i].nd],bot[i].nd));
		 	//将当前更新的数据放入小根堆
		 }
	}
}
inline int read(){
	int x=0,w=1;char ch=0;
	while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
	if(ch=='-') w=-1,ch=getchar();
	while(ch>='0'&&ch<='9')x=x*10+ch-48,ch=getchar();
	return x*w;
}
int main(){
	n=read();m=read();s=read();
	int x,y,w;
	for(int i=1;i<=m;i++){
		x=read();y=read();w=read();
		add(x,y,w);
	}
	dijkstra(s);
	for(int i=1;i<=n;i++) 
	 if(dis[i]!=2147483647)printf("%d ",dis[i]);
	 else cout<<2147483647<<" ";
} 

第二种版本:

#include
#include
#include
#include
#include
#include
using namespace std;
struct arr{
    int nd,nx,co;
}bot[500005];
struct node{//重载运算符,可以上网搜
    int u,d;
    bool operator <(const node& r) const{
        return d>r.d;
    }
};
int n,m,s,cnt;
int head[150000],dis[150000],vis[150000];
inline int read(){
    int x=0,w=1;char ch;
    while(ch!='0'&&(ch<'0'||ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+(ch-48),ch=getchar();
    return x*w;
}
inline void add(int u,int v,int w){ bot[++cnt].nd=v; bot[cnt].nx=head[u]; bot[cnt].co=w; head[u]=cnt; }
inline void dijkstra(){
    for(register int i=1;i<=n;++i) dis[i]=2147483647;
    dis[s]=0;vis[s]=1;
    priority_queue<node> q;
    q.push((node){s,0});
    while(!q.empty()) {
        node now=q.top();q.pop();
        int u=now.u,d=now.d;vis[u]=0;
        for(register int i=head[u];i;i=bot[i].nx) {
            int v=bot[i].nd,w=bot[i].co;
            if(dis[u]+w<dis[v]) {
                dis[v]=dis[u]+w;
                if(!vis[v]) {vis[v]=1;q.push((node){v,dis[v]});}//放入堆中
            }
        }
    }
}
int main(){
    n=read();m=read();s=read();
    for(register int i=1;i<=m;++i) { int u=read(),v=read(),w=read(); add(u,v,w); }
    dijkstra();
    for(register int i=1;i<=n;++i) printf("%d ",dis[i]);
    return 0;
}

第三种版本:(三种版本主要是开的STL中的堆的写法不同,本蒟感觉下面这种好记些)

#include
#include
#include
#include
#include
#include
using namespace std;
struct arr{
	int nd,nx,co;
}bot[500005];

struct node {
	int u,d;
};
bool operator < (node a,node b) {//重载时注意一下里面的符号是反的
	if(a.d==b.d) return a.u>b.u;
	else return a.d>b.d;
}
priority_queue<node>q;

int n,m,s,cnt;
int head[150000],dis[150000],vis[150000];
inline int read(){
	int x=0,w=1;char ch;
	while(ch!='0'&&(ch<'0'||ch>'9')) ch=getchar();
	if(ch=='-') w=-1,ch=getchar();
	while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+(ch-48),ch=getchar();
	return x*w;
}
inline void add(int u,int v,int w){ bot[++cnt].nd=v; bot[cnt].nx=head[u]; bot[cnt].co=w; head[u]=cnt; }
inline void dijkstra() {
	//priority_queueq;
	dis[s]=0;vis[s]=1;q.push((node){s,0});
	while(!q.empty()) {
		node x=q.top(); q.pop();
		int u=x.u;vis[u]=0;
		if(x.d!=dis[u]) continue;
		for(register int i=head[u];i;i=bot[i].nx) {
			int v=bot[i].nd;
			if(dis[v]>dis[u]+bot[i].co) {
				dis[v]=dis[u]+bot[i].co;
	
				//if(!vis[v]) {
				//	vis[v]=1;
					q.push((node){v,dis[v]});
				//}
			}
		}
	}
}
int main() {
	freopen("P4779.in","r",stdin);
	freopen("P4779.out","w",stdout);
	n=read();m=read();s=read();
	memset(dis,0x3f,sizeof(dis));
	for(register int i=1;i<=m;++i) { int u=read(),v=read(),w=read();add(u,v,w);}
	dijkstra();
	for(register int i=1;i<=n;++i) printf("%d ",dis[i]);
	return 0;
}

附带朴素版(本人以前的一个博客)

###普通版

你可能感兴趣的:(模板,有向无环图,总结,图论)