求各位大佬帮忙查一下错误,刚学图论和倍增,自己也找不出错误QAQ
题目链接:点我
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
using namespace std;
long long n,m,cnt,adr,ans,ANS=2147483647,f[100050][26],g[100050][26],depth[100050];
vector <long long> son[100050];
struct Edge {
long long x,y,q;
}e[300050];
struct Bug {
int f,q;
}fa[100050];
bool operator < (Edge x,Edge y) {
return x.q<y.q;
}
long long get_father(long long x) {
if(x==fa[x].f) {
return x;
}
return fa[x].f=get_father(fa[x].f);
}
void Merge(long long x,long long y) {
int fx=get_father(x),fy=get_father(y);
if(fx!=fy) {
fa[fx].f=fy;
return ;
}
return ;
}
void dfs(int now) {
for(int i=0;i<son[now].size();i++) {
depth[i]=depth[now]+1;
for(int j=1;j<=25;j++) {
f[i][j]=f[f[i][j-1]][j-1]+f[i][j-1];
g[i][j]=max(g[i][j-1],g[f[i][j-1]][j-1]);
}
dfs(i);
}
}
long long LCA(int p1,int p2,int x) {
if(depth[p1]<depth[p2]) {
swap(p1,p2);
}
long long ans=-1;
for(long long i=25;i>=0;i--) {
if(depth[f[p1][i]]>=depth[p2]) {
if(g[p1][i]!=e[x].q) {
ans=max(ans,g[p1][i]);
}
p1=f[p1][i];
}
}
if(p1==p2) {
return ans;
}
for(int i=25;i>=0;i--) {
if(f[p1][i]!=f[p2][i]) {
if(g[p1][i]!=e[x].q&&g[p2][i]!=e[x].q) {
ans=max(ans,max(g[p1][i],g[p2][i]));
}
else if(g[p1][i]!=e[x].q) {
ans=max(ans,g[p1][i]);
}
else {
ans=max(ans,g[p2][i]);
}
p1=f[p1][i];
p2=f[p2][i];
}
}
return max(ans,max(g[p1][0],g[p2][0]));
}
int main() {
scanf("%lld%lld",&n,&m);
for(long long i=1;i<=m;i++) {
scanf("%lld%lld%lld",&e[i].x,&e[i].y,&e[i].q);
}
for(int i=1;i<=n;i++) {
fa[i].f=i;
}
sort(e+1,e+m+1);
for(int i=1;i<=m;i++) {
long long fx=get_father(e[i].x),fy=get_father(e[i].y);
if(fx!=fy) {
long long emax=max(e[i].x,e[i].y),emin=min(e[i].x,e[i].y);
Merge(emax,emin);
son[emin].push_back(emax);
fa[emax].f=emin;
fa[emax].q=e[i].q;
ans+=e[i].q;
++cnt;
}
if(cnt==n-1) {
adr=i+1;
break;
}
}
depth[1]=1;
for(long long i=1;i<=n;i++) {
f[i][0]=fa[i].f;
g[i][0]=fa[i].q;
}
dfs(1);
for(long long i=adr;i<=m;i++) {
ANS=min(ANS,e[i].q-LCA(e[i].x,e[i].y,i));
}
printf("%lld",ans+ANS);
return 0;
}