参加会议的最多员工数

一个公司准备组织一场会议,邀请名单上有 n 位员工。公司准备了一张 圆形 的桌子,可以坐下 任意数目 的员工。

员工编号为 0n - 1 。每位员工都有一位 喜欢 的员工,每位员工 当且仅当 他被安排在喜欢员工的旁边,他才会参加会议。每位员工喜欢的员工 不会 是他自己。

给你一个下标从 0 开始的整数数组 favorite ,其中 favorite[i] 表示第 i 位员工喜欢的员工。请你返回参加会议的 最多员工数目

示例 1:

img

输入:favorite = [2,2,1,2]
输出:3
解释:
上图展示了公司邀请员工 0,1 和 2 参加会议以及他们在圆桌上的座位。
没办法邀请所有员工参与会议,因为员工 2 没办法同时坐在 0,1 和 3 员工的旁边。
注意,公司也可以邀请员工 1,2 和 3 参加会议。
所以最多参加会议的员工数目为 3 。

示例 2:

输入:favorite = [1,2,0]
输出:3
解释:
每个员工都至少是另一个员工喜欢的员工。所以公司邀请他们所有人参加会议的前提是所有人都参加了会议。
座位安排同图 1 所示:
- 员工 0 坐在员工 2 和 1 之间。
- 员工 1 坐在员工 0 和 2 之间。
- 员工 2 坐在员工 1 和 0 之间。
参与会议的最多员工数目为 3 。

示例 3:

img

输入:favorite = [3,0,1,4,1]
输出:4
解释:
上图展示了公司可以邀请员工 0,1,3 和 4 参加会议以及他们在圆桌上的座位。
员工 2 无法参加,因为他喜欢的员工 0 旁边的座位已经被占领了。
所以公司只能不邀请员工 2 。
参加会议的最多员工数目为 4 。

提示:

  • n == favorite.length

  • 2 <= n <= 105

  • 0 <= favorite[i] <= n - 1

  • favorite[i] != i

class Solution {
​
    public int maximumInvitations(int[] favorite) {
        int n = favorite.length;
        List<List<Integer>> graph = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            graph.add(new ArrayList<>());
        }
        int[] in = new int[n];// 获取所有节点入度
        for (int i = 0; i < n; i++) {
            graph.get(i).add(favorite[i]);
            in[favorite[i]]++;
        }
        LinkedList<Integer> queue = new LinkedList<>();// 把入度为零的节点加入队列
        for (int i = 0; i < n; i++) {
            if (in[i] == 0) {
                queue.addLast(i);
            }
        }
        while (!queue.isEmpty()) {// 队列为空后,如果还有入度不为零的节点,他们将成环
            int cur = queue.removeFirst();
            for (int next : graph.get(cur)) {
                if (--in[next] == 0) {
                    queue.addLast(next);
                }
            }
        }
        int ans = 0;
        boolean[] visited = new boolean[n];
        HashMap<Integer, Integer> map = new HashMap<>();// 获取双向奔赴的两个节点
        for (int i = 0; i < n; i++) {
            if (in[i] == 1 && !visited[i]) {
                int cur = i;
                int cnt = 0;
                while (!visited[cur]) {
                    cnt++;
                    visited[cur] = true;
                    cur = favorite[cur];
                }
                if (cnt > 2) {
                    ans = Math.max(ans, cnt);// 如果当前环是多人首尾相连,那么答案可能在其中,取最大
                } else {
                    map.put(i, favorite[i]);// 双向奔赴
                }
            }
        }
        graph = new ArrayList<>();// 构建反图,为了获取从双向奔赴节点到最远端节点的距离
        for (int i = 0; i < n; i++) {
            graph.add(new ArrayList<>());
        }
        for (int i = 0; i < n; i++) {
            graph.get(favorite[i]).add(i);
        }
        int best = 0;// 所有双向奔赴节点都可以参与会议
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            int i = entry.getKey();
            int j = entry.getValue();
            // 从双向奔赴节点两侧各自宽度优先遍历,获取最大深度
            queue.add(i);
            int level = 0;
            while (!queue.isEmpty()) {
                int size = queue.size();
                while (size-- > 0) {
                    int cur = queue.removeFirst();
                    for (int next : graph.get(cur)) {
                        if (next != j) {
                            queue.addLast(next);
                        }
                    }
                }
                level++;
            }
            queue.add(j);
            while (!queue.isEmpty()) {
                int size = queue.size();
                while (size-- > 0) {
                    int cur = queue.removeFirst();
                    for (int next : graph.get(cur)) {
                        if (next != i) {
                            queue.addLast(next);
                        }
                    }
                }
                level++;
            }
            best += level;
        }
        return Math.max(ans, best);// 返回(多人首尾相连最大值) 与 (双向奔赴节点集群累加和)的最大值
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值