FAANGSWE算法与数据结构编程面试题-电影院座位安排

题目描述： 电影院有一个n x m的矩阵，表示影厅的座位安排。其中0表示座位为空，1表示座位已被占据。 现在有一部电影即将上映，要求你安排一下座位，安排规则如下：

• 组成一排的座位必须是连续的；
• 每组座位的大小必须是k个座位，且k是给定的正整数；
• 每排座位中最多只能有一个空座位。

示例： 给定一个3 x 4的矩阵matrix，k为2，要求安排座位使得座位空置尽量少。则一种可行的座位安排方式如下：

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

其中第二排的1和0可以组成一排大小为2的连续座位，并且只有一个空座位。

解题思路： 对于每一排座位，我们统计连续的1的个数以及连续1的开始位置和结束位置。 遍历所有的k大小的座位组合，在每个组合中寻找只有一个空座位的情况，并记录下来。 对于每一组只有一个空座位的情况，我们计算空座位的位置，然后修改原矩阵的对应位置。

代码示例：