LeetCode——情侣牵手

题目描述

N 对情侣坐在连续排列的 2N 个座位上,想要牵到对方的手。 计算最少交换座位的次数,以便每对情侣可以并肩坐在一起。 一次交换可选择任意两人,让他们站起来交换座位。

人和座位用 0 到 2N-1 的整数表示,情侣们按顺序编号,第一对是 (0, 1),第二对是 (2, 3),以此类推,最后一对是 (2N-2, 2N-1)。

这些情侣的初始座位 row[i] 是由最初始坐在第 i 个座位上的人决定的。

示例 1:
输入: row = [0, 2, 1, 3]
输出: 1
解释: 我们只需要交换row[1]和row[2]的位置即可。

示例 2:
输入: row = [3, 2, 0, 1]
输出: 0
解释: 无需交换座位,所有的情侣都已经可以手牵手了。

说明:
len(row) 是偶数且数值在 [4, 60]范围内。
可以保证row 是序列 0...len(row)-1 的一个全排列。

并查集(From 《挑战程序设计竞赛》)

并查集是一种处理元素分组情况的数据结构。并查集可进行如下操作:

  • 查询两个元素是否属于同一组。
  • 合并两个元素所在的组。

并查集使用树形结构:每个组对应一颗树,每个元素对应一个节点。哪个节点作为根节点无需关注,整体组成一棵树即可。

相关操作:

  • 初始化:每个节点作为一棵树。
  • 合并:从一颗树的根向另一颗树的根连边,合并为一棵树,即一棵树作为另一棵树的子树
  • 查询:沿着树向上走,查询包含这个元素的根节点。如果两个元素具有相同的根节点,则两个元素属于同一组。
  • 优化:
    • 合并时,高度小的树作为高度大的树的子树。确保最终的树的高度尽可能小。
    • 路径压缩:所有节点直接指向根节点,即树的高度为2。

并查集的实现

class Union{
    // 每个元素的根
    private int[] parent;
    // 并查集中分组个数
    private int count;

    // 初始化
    Union(int count){
        this.count=count;
        parent=new int[count];
        for(int i=0;i<count;++i){
            parent[i]=i;
        }
    }

    // 查询元素所在树的根
    public int findParent(int n){
        // 自己是自己的parent,即:自己的树的根
        while(n!=parent[n]){
            parent[n]=parent[parent[n]];// 压缩:同一集合中的所有元素指向同一个parent。
            n=parent[n];
        }
        return n;
    }

    // 合并元素x和y所属的组
    public void union(int x,int y){
        int parentX=findParent(x);
        int parentY=findParent(y);
        // 已在同一组
        if(parentX==parentY){
            return;
        }
        // 合并
        parent[parentX]=parentY;
        --count;// 总分组数
    }
}

本题解题思路(参考

坐错位置的情况与最少需要交换次数:

  • 1对情侣、2个座位,不需要交换。
  • 2对情侣、4个座位,交换1次。
  • 3对情侣、6个座位。首先交换1次使得其中1对情侣坐在一起,剩下2对情侣、4个座位。即需要交换2次。
  • 以此类推,得到f(n)=n-1。即:n对情侣相互坐错位置,最少需要交换n-1次。

把相互坐错位置的情侣放在一组,组内有n对情侣就需要n-1次交换。将N对情侣分为K组:n1,n2...nk,有n1+n2+...+nk=N。需要交换的次数分别为:n1-1、n2-1、...、nk-1,则总的最少交换次数为n1-1+n2-1+...+nk-1=n1+n2+...+nk-k=N-k。则,问题变为:N对情侣,根据相互坐错位置的条件分组,共有多少个分组

class Solution {
    public int minSwapsCouples(int[] row){
        int maxCount=row.length/2;// 情侣对数
        Union union=new Union(maxCount);
        for(int i=0;i<row.length-1;i+=2){
        // 除2表示属于哪一对情侣(0...n-1)
            union.union(row[i]/2,row[i+1]/2);
        }
        return maxCount-union.count;
    }
    class Union{
        private int[] parent;
        private int count;
        Union(int count){
            this.count=count;
            parent=new int[count];
            for(int i=0;i<count;++i){
                parent[i]=i;
            }
        }

        public int findParent(int n){
            while(n!=parent[n]){
                parent[n]=parent[parent[n]];
                n=parent[n];
            }
            return n;
        }

        public void union(int x,int y){
           int parentX=findParent(x);
            int parentY=findParent(y);
            if(parentX==parentY){
                return;
            }
            parent[parentX]=parentY;
            --count;// 总分组数
        }
    }
}
全部评论

相关推荐

11-15 18:39
已编辑
西安交通大学 Java
全村最靓的仔仔:卧槽,佬啥bg呢,本也是西交么
点赞 评论 收藏
分享
字节 飞书绩效团队 (n+2) * 15 + 1k * 12 + 1w
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务