LeetCode 题解工作台

最多可达成的换楼请求数目

我们有 n 栋楼,编号从 0 到 n - 1 。每栋楼有若干员工。由于现在是换楼的季节,部分员工想要换一栋楼居住。 给你一个数组 requests ,其中 requests[i] = [from i , to i ] ,表示一个员工请求从编号为 from i 的楼搬到编号为 to i 的楼。 一开始…

category

4

题型

code_blocks

7

代码语言

hub

3

相关题

当前训练重点

困难 · 回溯·pruning

bolt

答案摘要

我们注意到,换楼请求列表长度不超过 ,因此我们可以使用二进制枚举的方法枚举所有的换楼请求列表。具体地,我们可以使用一个长度为 的二进制数来表示一种换楼请求列表,其中第 位为 表示第 个换楼请求被选中,为 表示第 个换楼请求不被选中。 我们在 $[1, 2^{m})$ 的范围内枚举所有的二进制数,对于每个二进制数 ,我们先算出它的二进制表示中有多少个 ,记为 ,如果 比当前答案 大,…

Interview AiBox logo

Interview AiBox 实时 AI 助手,陪你讲清 回溯·pruning 题型思路

试试 AiBox 面试助手arrow_forward
description

题目描述

我们有 n 栋楼,编号从 0 到 n - 1 。每栋楼有若干员工。由于现在是换楼的季节,部分员工想要换一栋楼居住。

给你一个数组 requests ,其中 requests[i] = [fromi, toi] ,表示一个员工请求从编号为 fromi 的楼搬到编号为 toi 的楼。

一开始 所有楼都是满的,所以从请求列表中选出的若干个请求是可行的需要满足 每栋楼员工净变化为 0 。意思是每栋楼 离开 的员工数目 等于 该楼 搬入 的员工数数目。比方说 n = 3 且两个员工要离开楼 0 ,一个员工要离开楼 1 ,一个员工要离开楼 2 ,如果该请求列表可行,应该要有两个员工搬入楼 0 ,一个员工搬入楼 1 ,一个员工搬入楼 2 。

请你从原请求列表中选出若干个请求,使得它们是一个可行的请求列表,并返回所有可行列表中最大请求数目。

 

示例 1:

输入:n = 5, requests = [[0,1],[1,0],[0,1],[1,2],[2,0],[3,4]]
输出:5
解释:请求列表如下:
从楼 0 离开的员工为 x 和 y ,且他们都想要搬到楼 1 。
从楼 1 离开的员工为 a 和 b ,且他们分别想要搬到楼 2 和 0 。
从楼 2 离开的员工为 z ,且他想要搬到楼 0 。
从楼 3 离开的员工为 c ,且他想要搬到楼 4 。
没有员工从楼 4 离开。
我们可以让 x 和 b 交换他们的楼,以满足他们的请求。
我们可以让 y,a 和 z 三人在三栋楼间交换位置,满足他们的要求。
所以最多可以满足 5 个请求。

示例 2:

输入:n = 3, requests = [[0,0],[1,2],[2,1]]
输出:3
解释:请求列表如下:
从楼 0 离开的员工为 x ,且他想要回到原来的楼 0 。
从楼 1 离开的员工为 y ,且他想要搬到楼 2 。
从楼 2 离开的员工为 z ,且他想要搬到楼 1 。
我们可以满足所有的请求。

示例 3:

输入:n = 4, requests = [[0,3],[3,1],[1,2],[2,0]]
输出:4

 

提示:

  • 1 <= n <= 20
  • 1 <= requests.length <= 16
  • requests[i].length == 2
  • 0 <= fromi, toi < n
lightbulb

解题思路

方法一:二进制枚举

我们注意到,换楼请求列表长度不超过 1616,因此我们可以使用二进制枚举的方法枚举所有的换楼请求列表。具体地,我们可以使用一个长度为 1616 的二进制数来表示一种换楼请求列表,其中第 ii 位为 11 表示第 ii 个换楼请求被选中,为 00 表示第 ii 个换楼请求不被选中。

我们在 [1,2m)[1, 2^{m}) 的范围内枚举所有的二进制数,对于每个二进制数 maskmask,我们先算出它的二进制表示中有多少个 11,记为 cntcnt,如果 cntcnt 比当前答案 ansans 大,那么我们再判断 maskmask 是否是一个可行的换楼请求列表。如果是,那么我们就用 cntcnt 更新答案 ansans。判断 maskmask 是否是一个可行的换楼请求列表,只需要判断对于每个楼,它的净流入量是否为 00 即可。

时间复杂度 O(2m×(m+n))O(2^m \times (m + n)),空间复杂度 O(n)O(n)。其中 mmnn 分别是换楼请求列表的长度和楼的数量。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class Solution:
    def maximumRequests(self, n: int, requests: List[List[int]]) -> int:
        def check(mask: int) -> bool:
            cnt = [0] * n
            for i, (f, t) in enumerate(requests):
                if mask >> i & 1:
                    cnt[f] -= 1
                    cnt[t] += 1
            return all(v == 0 for v in cnt)

        ans = 0
        for mask in range(1 << len(requests)):
            cnt = mask.bit_count()
            if ans < cnt and check(mask):
                ans = cnt
        return ans
speed

复杂度分析

指标
时间O(2^(M * (M + N))
空间O(N)
psychology

面试官常问的追问

外企场景
  • question_mark

    Candidate demonstrates understanding of backtracking with pruning techniques.

  • question_mark

    Candidate effectively utilizes bit manipulation for optimization.

  • question_mark

    Candidate identifies early pruning conditions to limit unnecessary exploration.

warning

常见陷阱

外企场景
  • error

    Failure to implement proper pruning leads to excessive exploration of infeasible branches.

  • error

    Not optimizing state representation, which can lead to slow performance.

  • error

    Overlooking the importance of checking net transfers early during the backtracking process.

swap_horiz

进阶变体

外企场景
  • arrow_right_alt

    Consider optimizing the pruning strategy to handle larger inputs.

  • arrow_right_alt

    Explore dynamic programming approaches to compare with backtracking.

  • arrow_right_alt

    Investigate greedy methods for approximate solutions in specific cases.

help

常见问题

外企场景

最多可达成的换楼请求数目题解:回溯·pruning | LeetCode #1601 困难