1-11G. JM的月亮神树

1-11G. JM的月亮神树

题解

定义：一个图的不科学生成树是的一棵子树，在这棵子树上，从核心节点到任意一个节点的最短路径长度，和在原图上是等长的。其中节点是月亮神树的核心节点。

根据定义很容易发现，这棵树是由核心节点到其它任意一个节点的最短路所构成的。所以，我们可以先用Dijkstra将所有最短路给跑出来。

再根据每条边的编号当作权值对所有的最短路进行更新。

不过如果将所有最短路的边记录下来跑最小生成树的话不能保证到每个点的路径都是最短路，所以会出现WA的情况。

源码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<stack>
#include<map>
using namespace std;
#define MAXN 300005
typedef long long LL;
#define int LL
const LL INF=0x7f7f7f7f7f7f;
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
int n,m,s,head[MAXN],tot,cnt,sc;
struct ming{int u,v,w,id;}a[MAXN],b[MAXN];
struct edge{int to,nxt,id,paid;}e[MAXN<<1];
void addEdge(int u,int v,int w,int i){e[++tot]=(edge){v,head[u],i,w};head[u]=tot;} 
LL dis[MAXN],sval,did[MAXN];
bool vis[MAXN],used[MAXN];
vector<int>G[MAXN];
struct step{
	int id;LL t;
	bool friend operator < (const step &x,const step &y){
		return x.t>y.t;
	}
};
priority_queue<step> q;
void Dijkstra(int u){
	while(!q.empty())q.pop();
	for(int i=1;i<=n;i++)dis[i]=INF;
	q.push((step){u,0});dis[u]=0;
	while(!q.empty()){
		int t=q.top().id;q.pop();
		if(vis[t])continue;vis[t]=1;
		for(int i=head[t];i;i=e[i].nxt){
			int v=e[i].to;
			if(dis[v]>dis[t]+1ll*e[i].paid)
				dis[v]=dis[t]+1ll*e[i].paid,did[v]=0;
				q.push((step){v,dis[v]});
			if(dis[v]==dis[t]+1ll*e[i].paid){
				if(!did[v])did[v]=e[i].id;
				if(e[i].paid<a[did[v]].w)did[v]=e[i].id;
				else if(e[i].paid==a[did[v]].w&&e[i].id<did[v])
					did[v]=e[i].id;
			}
		}
	}
}
bool cmp(ming x,ming y){
	if(x.w==y.w)return x.id<y.id;
	return x.w<y.w;
}
signed main(){
	//freopen("moontree.in","r",stdin);
	//freopen("moontree.out","w",stdout);
	read(n);read(m);read(s);
	for(int i=1;i<=m;i++){
		int u,v,w;read(u);read(v);read(w);
		addEdge(u,v,w,i);addEdge(v,u,w,i); 
		a[i]=(ming){u,v,w,i};
	}
	//printf("36:%d %d %d\n",a[36].u,a[36].v,a[36].w);
	//printf("44:%d %d %d\n",a[44].u,a[44].v,a[44].w);
	Dijkstra(s);
	for(int i=1;i<=n;i++){
		int siz=G[i].size();
		for(int j=0;j<siz;j++)
			b[++cnt]=a[G[i][j]];
	}
	sort(b+1,b+cnt+1,cmp);
	for(int i=1;i<=n;i++)
		if(did[i]){
			sc++;sval+=1ll*a[did[i]].w;
			used[did[i]]=1;
		}
	printf("%d %lld\n",sc,sval);
	for(int i=1;i<=m;i++)if(used[i])printf("%d ",i);
	return 0;
}

谢谢！！！