AtCoder Beginner Contest 233 F - Swap and Sort(并查集+dfs)

题目描述:

我们有一个关于(1—N)的全排列P;
现在给出m次操作。
每次操作包含x和y。该操作可以交换Px与Py的值。
现在问你是否可以通过最多执行 5×105次操作对 P 进行排序
操作可以交换顺序。
如果可能,请给出一个这样的操作序列。 否则输出-1。

题解:

我们将交换操作抽象成图,这样我们根据并查集判断i与p[i]是否在同一联通分量之中,如果所有i与对应p[i]都可以在同一联通分量找到,我们就开始匹配。先找到叶子节点,因为只有先将叶子节点固定了,我们才能够操作他的父节点。那用两次dfs就可以了。

#include<bits/stdc++.h>
using namespace std;
const int N=1e3+5;
typedef long long LL;
typedef pair<int,int>pii;
int p[N],res=0;
bool vis[N];
vector<pii>G[N];
vector<int>ans;
int pre[N];
int findx(int x)
{
    return x==pre[x]?pre[x]:pre[x]=findx(pre[x]);
}
bool merge(int x,int y)
{
    x=findx(x),y=findx(y);
    if(x==y) return 0;
    else pre[x]=y;
    return 1;
}
int match(int u,int v,int fa)
{
    if(p[u]==v)return 1;//匹配成功
    for(auto i:G[u])
    {
        int x=i.first,id=i.second;
        if(x==fa)continue;
        if(match(x,v,u))
        {
            swap(p[u],p[x]);//交换p[u],p[x];
            ans.push_back(id);//答案中加入操作编号。
            return 1;
        }
    }
    return 0;
}

void dfs(int u)
{
    vis[u]=1;
    for(auto i:G[u])
    {
        int v=i.first;
        if(!vis[v]) dfs(v);//先递归找子节点,先将子节点固定
    }
    match(u,u,-1);
}
int main()
{
    int n,m;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        pre[i]=i;
        scanf("%d",&p[i]);
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        if(merge(a,b))//如果在同一集合里就不用再加边了。
        {
            G[a].push_back({b,i});//加边的时候不要忘了加入操作序号
            G[b].push_back({a,i});
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(findx(i)==findx(p[i]))continue;//如果i与p[i]不在同一集合里就退出
        else
        {
            puts("-1");
            return 0;
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(!vis[i])dfs(i);
    }
    printf("%d\n",ans.size());
    for(auto i:ans)printf("%d ",i);
}

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值