注意:
- 注意在求严格次小生成树时,不能只预处理两点之间最大的树边,因为当最大树边和当前枚举的非树边长度相同时,就不能替换了,但此时却可以替换长度次大的树边。因此还需同时预处理出长度次大的树边。
#include<algorithm>
#include<string.h>
using namespace std;
using namespace std;
typedef long long ll;
const int N=550,M=10010;
int n,m;
struct node{
int a,b,w;
bool f;
bool operator < (const node &t)const{
return w<t.w;
}
}edges[M];
ll res,num;
int dis1[N][N],dis2[N][N];
int p[N],ne[M],e[M],h[N],w[M],idx;
int find(int x)
{
return x==p[x]?x:p[x]=find(p[x]);
}
void add(int a,int b,int c)
{
e[idx]=b;
w[idx]=c;
ne[idx]=h[a];
h[a]=idx++;
}
void dfs(int u,int father,int d1,int d12,int dist1[],int dist2[])
{
dist1[u]=d1,dist2[u]=d12;
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(j==father) continue;
int td1=d1,td2=d12; //需要用变量接一下值,因为有好多点链接父节点,多次用到
if(w[i]>td1){
td2=td1,td1=w[i];
}
else if(w[i]<td1 && w[i]>td2)
td2=w[i];
dfs(j,u,td1,td2,dist1,dist2);
}
}
int main()
{
ios::sync_with_stdio(false);
cin>>n>>m;
memset(h,-1,sizeof h);
for(int i=0;i<m;i++)
{
int a,b,w;
cin>>a>>b>>w;
edges[i]={a,b,w,false};
}
sort(edges,edges+m);
for(int i=0;i<=n;i++) p[i]=i;
for(int i=0;i<m;i++)
{
int a=edges[i].a,b=edges[i].b,w=edges[i].w;
int pa=find(a),pb=find(b);
if(pa!=pb)
{
num+=w;
p[pa]=pb;
edges[i].f=true;
add(a,b,w),add(b,a,w);
}
}
//搞出dis
for(int i=1;i<=n;i++) dfs(i,-1,0,0,dis1[i],dis2[i]);
res=1e18;
for(int i=0;i<m;i++)
{
if(edges[i].f==false)
{
int a=edges[i].a;
int b=edges[i].b;
int w=edges[i].w;
ll t;
if(w>dis1[a][b]) t=num-dis1[a][b]+w;
else if(w==dis1[a][b]&&w>dis2[a][b]) t=num-dis2[a][b]+w;
res=min(res,t);
}
}
cout<<res<<endl;
return 0;
}
`
```