60pts,kruskal,知道哪错了玄关求改
查看原帖
60pts,kruskal,知道哪错了玄关求改
749811
Licis_Subway楼主2024/2/22 22:27

先贴代码:

#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=yx=y的情况就算不对,就是说如果xx直接可以上提到yy或者yy直接可以上提到xx的 情况算不对。

总是会算多。

玄关!有没有大佬帮我一下

2024/2/22 22:27
加载中...