求助
查看原帖
求助
525375
Richard_Whr楼主2023/9/23 18:52

和别人思路不太一样,但是并不知道哪里错了

如果同排两个数一样,那么就连一条边,表示这两个里面必须换一个且只能换一个

如果不同排有两个相同的数,则加入并查集,表示这两个如果换一个必须换另外一个

这样处理,每个点上都有一个权值,即所在并查集的size

然后dp一下,最后累加

为啥挂了50

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10,M=5e5+10;
int h[N],e[M],ne[M],w[N],idx;
int sta[N],stb[N];
int a[N],b[N];
int p[N],sz[N];
int n;
int f[N][2];
bool st[N];

void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}

void dfs(int u,int fa)
{
    st[u]=true;
    f[u][0]=0,f[u][1]=w[u];
    for(int i=h[u];~i;i=ne[i])
    {
        int son=e[i];
        if(st[son]||son==fa) continue;
        
        dfs(son,u);
        f[u][0]+=f[son][1];
        f[u][1]+=f[son][0];
    }
}

int main()
{
    memset(h,-1,sizeof h);
    scanf("%d",&n);
    
    for(int i=1;i<=n;i++) p[i]=i,sz[i]=1;
    
    for(int i=1;i<=n;i++) 
    {
        scanf("%d",&a[i]);
        if(sta[a[i]])
        {
            int x=sta[a[i]];
            add(x,i),add(i,x);
        }
        else sta[a[i]]=i;
        
        if(stb[a[i]])
        {
            int x=stb[a[i]];
            int pa=find(x),pb=find(i);
            sz[pb]+=sz[pa];
            p[pa]=pb;
        }
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&b[i]);
        if(stb[b[i]])
        {
            int x=stb[b[i]];
            add(x,i),add(i,x);
        }
        else stb[b[i]]=i;
        
        if(sta[b[i]])
        {
            int x=sta[b[i]];
            int pa=find(x),pb=find(i);
            sz[pb]+=sz[pa];
            p[pa]=pb;
        }
    }
    
    for(int i=1;i<=n;i++) w[i]=sz[find(i)];
    
    memset(f,0x3f,sizeof f);
    int res=0;
    for(int i=1;i<=n;i++)
    {
        if(!st[i])
        {
            dfs(i,-1);
            res+=min(f[i][0],f[i][1]);
        }
    }
    
    printf("%d",res);
    
    return 0;
}

2023/9/23 18:52
加载中...