给你一个 m * n 的矩阵 seats 表示教室中的座位分布。如果座位是坏的(不可用),就用 '#' 表示;否则,用 '.' 表示。
学生可以看到左侧、右侧、左上、右上这四个方向上紧邻他的学生的答卷,但是看不到直接坐在他前面或者后面的学生的答卷。请你计算并返回该考场可以容纳的一起参加考试且无法作弊的最大学生人数。
学生必须坐在状况良好的座位上。
示例 1:
输入:seats = [["#",".","#","#",".","#"],
[".","#","#","#","#","."],
["#",".","#","#",".","#"]]
输出:4
解释:教师可以让 4 个学生坐在可用的座位上,这样他们就无法在考试中作弊。
题解:
状态压缩每一行的状态["#",".","#","#",".","#"]可以为二进制的010010即十进制18。
每行的总状态数则为2^n-1
构造dp数组dp[i][j]表示第i行状态为j时的最大人数,如果状态j不合法(判断方法如下)则人数定为0。
如何判断某状态的合理性:如果坐人的位置是坏的则直接不合理。若某行的状态为i,将i左右移一位分别与i且运算,存在一侧大于0则不合理。
如何判断某状态结合上一行状态的合理性:同理,如果坐人的位置是坏的则直接不合理。若某行的状态为i,上一行为j,将i左右移一位分别与j且运算,存在一侧大于0则不合理。
很容易的可以得到状态转移方法,给出伪代码如下:
if j状态不合理:
dp[i][j] = 0
return
max = 0
for k in 0..2^n-1:
if j与上一行k匹配合理 && dp[i-1][k]大于max:
max = dp[i-1][k]
dp[i][j] = max
以下是全部代码:dp可以优化为一维数组,降低空间复杂度。
// 计算input状态的人数
int getSize(int input) {
int output = 0;
while (input != 0) {
if (input % 2 == 1) output++;
input = input >> 1;
}
return output;
}
// 错位判断合理性
bool isWrong(int a, int b) {
return (((a << 1) & b) > 0) || (((a >> 1) & b) > 0);
}
// dp[i][j] 第i行为j状态时的最大人数
int maxStudents(vector<vector<char>> &seats) {
// m行 * 2^n-1(每行状态数) * 2^n-1
int m = seats.size();
int n = seats[0].size();
int nn = pow(2, n) - 1;
int **dp = new int *[m];
for (int i = 0; i < m; ++i) dp[i] = new int[nn];
int *state = new int[m];
for (int i = 0; i < m; ++i) {
int temp = 0;
for (int j = 0; j < n; ++j) {
if (seats[i][j] == '.') temp += (1 << j);
}
state[i] = temp;
}
for (int i = 0; i < nn; ++i) {
if (((i | state[0]) > state[0]) || isWrong(i, i)) {
dp[0][i] = 0;
continue;
}
dp[0][i] = getSize(i);
}
for (int i = 1; i < m; ++i) {
for (int j = 0; j < nn; ++j) {
if (((j | state[i]) > state[i]) || isWrong(j,j)) {
dp[i][j] = 0;
continue;
}
int tt = getSize(j);
int r = 0;
for (int k = 0; k < nn; ++k) {
int t = dp[i - 1][k];
if ((t == 0) || isWrong(j, k)) {
continue;
}
if (t > r) r = t;
}
dp[i][j] = r + tt;
}
}
int output = 0;
for (int i = 0; i < nn; ++i) {
int t = dp[m - 1][i];
if (t > output) output = t;
}
return output;
}