lightOJ 1002 最小瓶頸樹 ..

第一次要把最小生成樹保存下來。還是對算法理解有問題。一開始的時候要把起始點設置爲1，並把所有的相連邊加入隊列中。

自己寫edge的時候不喜歡記錄來源，所以用了個Pair，實在捉雞，可以直接寫成帶有from的struct，這樣代碼複雜度會減少很多。

#include<stdio.h>
#include<vector>
#include<algorithm>
#include<string.h>
#include<queue>
#include<utility>
using namespace std;
const int INF=0x3f3f3f3f;
struct edge
{
	int to;
	int cost;
	bool operator < (const edge &o) const
	{
		return cost<o.cost;
	}
}e[20000];

typedef pair<edge,int > P;
vector<edge> G[600];

vector<edge> minTree[600];

bool used[500];

int nv,ne;

priority_queue<P,vector<P>,greater<P> > q;

void AddEdge(vector<edge>* GG,int f,int t,int c)
{
	edge te;
	te.to=t;
	te.cost=c;
	GG[f].push_back(te);
	edge te2;
	te2.to=f;
	te2.cost=c;
	GG[t].push_back(te2);
}

bool SetMinTree(int t)
{
	used[t]=1;
	for(int i=0;i<G[t].size();i++)
	{
		edge te=G[t][i];	
		q.push(P(te,t));
	}
	while(!q.empty())
	{
		P tp=q.top();
		q.pop();
		int f=tp.second;
		edge te=tp.first;
		if(!used[te.to])
		{
			used[te.to]=1;
			AddEdge(minTree,f,te.to,te.cost);
			for(int i=0;i<G[te.to].size();i++)
			{
				edge tte=G[te.to][i];
				if(!used[tte.to])
				{
					q.push(P(tte,te.to));
				}
			}	
		}
	}
	for(int i=0;i<nv;i++)
	{
		if(!used[i])
			return false;
	}
	return true;
}

int ans[600];

void dfs(int max,int f)
{
	if(max<ans[f])
		ans[f]=max;
	for(int i=0;i<minTree[f].size();i++)
	{
		edge te=minTree[f][i];
		if(ans[te.to]==INF)
		{
			dfs(max>te.cost?max:te.cost,te.to);
		}
	}
}

void printG(vector<edge>* GG)
{
	for(int i=0;i<nv;i++)
	{
		printf("%d: ",i);
		for(int j=0;j<GG[i].size();j++)
		{
			printf("%d ",GG[i][j].to);
		}	
		printf("\n");
	}
}
int main()
{
	int cas;
	scanf("%d",&cas);
	for(int T=1;T<=cas;T++)
	{
		for(int i=0;i<600;i++)
		{
			G[i].clear();
			minTree[i].clear();
		}
		memset(used,0,sizeof(used));
		scanf("%d%d",&nv,&ne);
		for(int i=0;i<ne;i++)
		{
			int f,t,c;
			scanf("%d%d%d",&f,&t,&c);
			AddEdge(G,f,t,c);
		}
		int target;
		
	//	printG(G);
		
		scanf("%d",&target);
		if(ne>0)
			SetMinTree(target);
		
	//	printG(minTree);

		for(int i=0;i<600;i++)
			ans[i]=INF;
		dfs(0,target);
		printf("Case %d:\n",T);
		for(int i=0;i<nv;i++)
		{
			if(ans[i]==INF)
				printf("Impossible\n");
			else
				printf("%d\n",ans[i]);
		}
	}
	return 0;
}