UVA-10462-Is There A Second Way Left? （次小生成树带重边）

题目链接：UVA-10462-Is There A Second Way Left?

有重边的次小生成树。。。所以用了邻接表的Prim。。不过边很少用Kruskal应该也能过。

#include<bits/stdc++.h>
using namespace std;
int n,m;
const int INF=0x3f3f3f3f;
int Max[107][107];
bool vis[107],evis[207];
struct Edge{
	int u,v,d,id;
	Edge(){}
	Edge(int _u,int _v,int _d,int _id):u(_u),v(_v),d(_d),id(_id) {}
	bool operator < (const Edge & r) const
	{
		return d>r.d;
	}
};
vector<Edge> adj[107];
bool dfs(int u)
{
	queue<int> q;
	q.push(1);
	vis[1]=true;
	while(!q.empty())
	{
		int t=q.front();q.pop();
		for(int i=0;i<adj[t].size();i++)
		{
			int v=adj[t][i].v;
			if(vis[v])	continue;
			q.push(v);
			vis[v]=true;
		}
	}
}
bool iscon()
{
	for(int i=1;i<=n;i++)	if(!vis[i])	return false;
	return true;
}
int Prim()
{
	int res=0;
	memset(vis,0,sizeof(vis));
	memset(Max,0,sizeof(Max));
	memset(evis,0,sizeof(evis));
	priority_queue<Edge> q;
	vis[1]=true;
	for(int i=0;i<adj[1].size();i++)	q.push(adj[1][i]);
	while(!q.empty())
	{
		Edge e=q.top();q.pop();
		int u=e.u,v=e.v,d=e.d;
		if(vis[v])	continue;
		for(int i=1;i<=n;i++)
			if(vis[i])	Max[v][i]=Max[i][v]=max(Max[i][u],d);
		res+=d;
		evis[e.id]=true;
		vis[v]=true;
		for(int i=0;i<adj[v].size();i++)
		{
			if(vis[adj[v][i].v])	continue;
			q.push(adj[v][i]);
		}
	}
	int ans=0x3f3f3f3f;
	for(int u=1;u<=n;u++)
		for(int j=0;j<adj[u].size();j++)
		{
			int v=adj[u][j].v;
			if(evis[adj[u][j].id])	continue;
			ans=min(ans,res+adj[u][j].d-Max[u][v]);
		}
	return ans;
}
int main()
{
	int T,kase=0;
	scanf("%d",&T);
	while(T--)
	{
		int u,v,c;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)	adj[i].clear();
		for(int i=0;i<m;i++)
		{
			scanf("%d%d%d",&u,&v,&c);
			adj[u].push_back(Edge(u,v,c,i));
			adj[v].push_back(Edge(v,u,c,i));
		}
		memset(vis,0,sizeof(vis));
		dfs(1);
		printf("Case #%d : ",++kase);
		if(!iscon())	puts("No way");
		else if(n==m+1)	puts("No second way");
		else printf("%d\n",Prim());
	}
	return 0;
}