先贴代码:
#include<cstdio>
#include<vector>
#include<algorithm>
#include<unordered_set>
#include<cmath>
using namespace std;
struct Node
{
int u,v,w,id;
};
struct Mode
{
int v,w,id;
};
Node make_node(int u,int v,int w,int id)
{
Node ans;
ans.u=u;ans.v=v;ans.w=w;ans.id=id;
return ans;
}
Mode make_mode(int v,int w,int id)
{
Mode ans;
ans.v=v;ans.w=w;ans.id=id;
return ans;
}
bool cmp(Node x,Node y){return x.w>y.w;}
const int N=1e4+5;
int fa[N],dep[N],w[N][25],newfa[N][25],a[N][N];
bool vis[N];
vector<Node> edges;
unordered_set<int> used;
vector<Mode> G[N];
int find(int x)
{
if(x==fa[x]) return x;
else return fa[x]=find(fa[x]);
}
void Union(int x,int y)
{
int fx=find(fa[x]),fy=find(fa[y]);
fa[fx]=fy;
}
int kruskal(const int n)
{
int cnt=0,sum=0;
vector<Node>::iterator it=edges.begin();
for(;it!=edges.end();it++)
{
if(find(it->u)==find(it->v)) continue;
++cnt;sum+=it->w;
used.insert(it->id);
a[it->u][it->v]=a[it->v][it->w]=it->w;
if(it->id%2==0) used.insert(it->id-1);
else used.insert(it->id+1);
Union(it->u,it->v);
}
if(cnt==n-1) return sum;
else return -1;
}
inline int min(int x,int y){return x<=y?x:y;}
inline int lowbit(int x){return x&(-x);}
void dfs(int idx,int f)
{
newfa[idx][0]=f;
dep[idx]=dep[f]+1;
for(vector<Mode>::iterator it=G[idx].begin();it!=G[idx].end();it++)
{
if(!used.count(it->id)) continue;
if(vis[it->v]) continue;
vis[it->v]=true;
w[it->v][0]=it->w;
dfs(it->v,idx);
}
}
int LCA(int x,int y,const int n)
{
int ans=2e9;
if(dep[x]>dep[y]) std::swap(x,y);
int copyy=y;
for(int i=20;i>=0;i--)
if(dep[newfa[y][i]]>=dep[x])
{
ans=min(ans,w[y][i]);
y=newfa[y][i];
}
for(int i=20;i>=0;i--)
{
if(newfa[x][i]==newfa[y][i]) continue;
ans=min(ans,min(w[x][i],w[y][i]));
x=newfa[x][i];
y=newfa[y][i];
}
ans=min(ans,min(w[x][0],w[y][0]));
return ans;
}
int main()
{
int n,m,edgescnt=0;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) fa[i]=i;
for(int i=1;i<=m;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
edges.push_back(make_node(x,y,z,++edgescnt));
edges.push_back(make_node(y,x,z,++edgescnt));
G[x].push_back(make_mode(y,z,edgescnt-1));
G[y].push_back(make_mode(x,z,edgescnt));
}
sort(edges.begin(),edges.end(),cmp);
kruskal(n);
for(int i=1;i<=n;i++)
if(!vis[i]) dfs(i,i);
for(int j=1;j<=20;j++)
for(int i=1;i<=n;i++)
{
newfa[i][j]=newfa[newfa[i][j-1]][j-1];
w[i][j]=min(w[i][j-1],w[newfa[i][j-1]][j-1]);
}
int q;
scanf("%d",&q);
while(q--)
{
int x,y;
scanf("%d%d",&x,&y);
if(find(x)!=find(y)) printf("-1\n");
else printf("%d\n",LCA(x,y,n));
}
return 0;
}
我的错点是LCA里面如果在第一次for里面就找到x=y的情况就算不对,就是说如果x直接可以上提到y或者y直接可以上提到x的 情况算不对。
总是会算多。
玄关!有没有大佬帮我一下