在Python中，以下代码是如何工作的

Question

说明

越狱

监狱可以表示为牢房列表。每个牢房只关押一名囚犯。1表示未锁定的单元格，0表示锁定的单元格。

[1,1,0,0,1,0]

从最左边的牢房开始，你的任务是看看有多少囚犯可以被释放。你是第一间牢房里的囚犯。如果第一个单元格被锁定，则无法释放任何人。每次释放囚犯时，锁定的牢房就会解锁，解锁的牢房也会再次锁定。

因此，如果我们使用上面的例子：

[1, 1, 0, 0, 0, 1, 0]
# You free the prisoner in the 1st cell.
[0, 0, 1, 1, 1, 0, 1] 
# You free the prisoner in the 3rd cell (2nd one locked).
[1, 1, 0, 0, 0, 1, 0]
# You free the prisoner in the 6th cell (3rd, 4th and 5th locked).
[0, 0, 1, 1, 1, 0, 1]
# You free the prisoner in the 7th cell - and you are done!
Here, we have set free 4 prisoners in total.

创建一个函数，根据这种独特的监狱安排，返回获释囚犯的数量

示例

freed_prisoners([1, 1, 0, 0, 0, 1, 0]) ➞ 4
freed_prisoners([1, 1, 1]) ➞ 1
freed_prisoners([0, 0, 0]) ➞ 0
freed_prisoners([0, 1, 1, 1]) ➞ 0

票据

• 你是第一间牢房里的囚犯。你必须被释放才能释放其他人。

• 你必须释放一名囚犯才能换锁。因此，在第二个例子中，在释放第一个囚犯后，输入为[1，1，1]，锁变为[0，0，0]。既然所有的牢房都被锁上了，你就不能再释放囚犯了。

• 你总是从列表中最左边的元素(第一个牢房)开始。如果你右边的所有牢房都是零，你就不能再释放任何囚犯了。

学分：Edabit

我在解决方案中发现了以下代码：

def freed_prisoners(prison):
freed = [k for k, _ in groupby(prison)]
return len(freed) if freed[0] == 1 else 0

我知道groupby函数会给我密钥：[1, 0, 1, 0]。

我的问题是，这个代码是如何与逻辑匹配的？它只是碰巧给出了所需的相同答案吗？

Answer 1

从囚犯的角度来看，越狱是这样的：

1. 如果我的手机被锁定，没有人被释放(freed[0] == 1)。结束。否则继续执行步骤2
2. 我离开牢房后，锁会切换，我所在小组的任何人都会呆在他们的牢房里，因为牢房只是从解锁变成了锁定。跳过我组中的所有人(k, _ in groupby(prison))
3. 对单元块的其余部分重复步骤1和2

从第2步中，我们可以推断出，每个锁定状态相同的组中只有一名囚犯获释。我们还可以推断，从下面的每一组中，我们可以释放一名囚犯。因此，确定被释放的囚犯人数可以简化为计算不同群体的人数(如果我的牢房被锁，则为零)。这将是len(freed)。

从下面的代码中可以更容易地看到这一点。还要注意第二个例子；每个组仅释放一个"；部分与第一个例子相比：

import itertools
def freed_prisoners(prison):
groups = [(k,list(v)) for k,v in itertools.groupby(prison)]
print(f"Groups: {groups}")
freed = [k for k, _ in groups]
persons = len(freed) if freed[0] == 1 else 0
print(f"Freed list: {freed}; freed {persons} persons")
return persons
freed_prisoners([1, 1, 0, 0, 0, 1, 0])  # ➞ 4
freed_prisoners([1, 0, 1, 0])           # ➞ 4
freed_prisoners([1, 1, 1])              # ➞ 1
freed_prisoners([0, 0, 0])              # ➞ 0
freed_prisoners([0, 1, 1, 1])           # ➞ 0

输出

Groups: [(1, [1, 1]), (0, [0, 0, 0]), (1, [1]), (0, [0])]
Freed list: [1, 0, 1, 0]; freed 4 persons
Groups: [(1, [1]), (0, [0]), (1, [1]), (0, [0])]
Freed list: [1, 0, 1, 0]; freed 4 persons
Groups: [(1, [1, 1, 1])]
Freed list: [1]; freed 1 persons
Groups: [(0, [0, 0, 0])]
Freed list: [0]; freed 0 persons
Groups: [(0, [0]), (1, [1, 1, 1])]
Freed list: [0, 1]; freed 0 persons