1601. 最多可達成的換樓請求數目

• 題目描述
• 題解思路
• 題解代碼
• itertools.combinations()

題目描述

我們有 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

題解思路

我們從請求的 最長 長度開始倒序遍歷（并且進行排列組合）

直到第一個滿足條件的狀況出現 我們將其返回

題解代碼

class Solution:def maximumRequests(self, n: int, requests: List[List[int]]) -> int:for i in range(len(requests)):for comb in combinations(requests,i):cnt=[0]*nfor x,y in cmb:cnt[x]-=1cnt[y]+=1if all(not c for c in cnt):return ireturn 0

itertools.combinations()

itertools.combinations(iterable[, r]創建一個迭代器，返回iterable中所有長度為r的子序列，返回的子序列中的項按輸入iterable中的順序排序 (不帶重復).r 指定生成排列的元素的長度，如果不指定，則默認為可迭代對象的元素長度。

例子

from itertools import combinationslist(combinations('ABC', 2)) [('A', 'B'), ('A', 'C'), ('B', 'C')]

總結

以上是生活随笔為你收集整理的LeetCode刷题日记2022-2-28/1601. 最多可达成的换楼请求数目-排列组合的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。