给你一个 m x n 的字符矩阵 box ,它表示一个箱子的侧视图。箱子的每一个格子可能为:
- ‘#’ 表示石头
- ‘*’ 表示固定的障碍物
- ‘.’ 表示空位置
这个箱子被 顺时针旋转 90 度 ,由于重力原因,部分石头的位置会发生改变。每个石头会垂直掉落,直到它遇到障碍物,另一个石头或者箱子的底部。重力 不会 影响障碍物的位置,同时箱子旋转不会产生惯性 ,也就是说石头的水平位置不会发生改变。
题目保证初始时 box 中的石头要么在一个障碍物上,要么在另一个石头上,要么在箱子的底部。
请你返回一个_ _n x m的矩阵,表示按照上述旋转后,箱子内的结果。
先将数组旋转 90 度,获得新数组,再进行后续处理。
int m = box.length; // 1
int n = box[0].length; // 3
char[][] res = new char[n][m]; // 3 * 1
for (int row = 0; row < m; row++) {
for (int col = 0; col < n; col++) {
res[col][m - 1 - row] = box[row][col];
}
}
处理数据的过程,每一列从下往上遍历,需要记录当前可能会被放石头的位置,从下往上,
- 如果是石头(‘#’),就把石头移到记录的位置;
- 如果是障碍物(‘*’),就更新记录可能会被放石头的位置为它的上一行;
- 如果是空位(‘.’),不进行处理。
for (int col = 0; col < m; col++) { // 列
int p = n - 1;
for (int row = n - 1; row >= 0; row--) { // 行
switch (res[row][col]) {
case '#':
res[row][col] = '.';
res[p][col] = '#';
p--;
break;
case '*':
p = row - 1;
break;
default:
break;
}
}
}
也可以先处理数据,再旋转
需要注意的地方(没错,就是我写的 bug):
- 旋转 90°,可不是简单把行变成列,它是第一行变成了最后一列,以此类推,最后一行变成第一列。
- 移动石头的过程,需要将原来石头的位置变成空位,不然石头可能会变多哟。
- 上面一点中,至于先移石头还是先变空位,需要注意先后关系,每一列最后一个位置和障碍物上一行时,石头的位置可能没发生变化,后变空位石头就会变少。
完整代码:
class Solution {
public char[][] rotateTheBox(char[][] box) {
int m = box.length; // 1
int n = box[0].length; // 3
char[][] res = new char[n][m]; // 3 * 1
for (int row = 0; row < m; row++) {
for (int col = 0; col < n; col++) {
res[col][m - 1 - row] = box[row][col];
}
}
for (int col = 0; col < m; col++) { // 列
int p = n - 1;
for (int row = n - 1; row >= 0; row--) { // 行
switch (res[row][col]) {
case '#':
if (p != row) {
res[row][col] = '.';
res[p][col] = '#';
}
p--;
break;
case '*':
p = row - 1;
break;
default:
break;
}
}
}
return res;
}
}
后记:
-
switch 支持的数据类型
- 基本数据类型:char,byte,short,int
- 包装数据类型:Character,Byte,Short,Integer
- 枚举类型:Enum
- 字符串类型:String(JDK 7+ 开始支持)
-
多分支 if 和 switch 的选择
- switch 支持的数据类型的限制
- 代码简洁性
- switch 只能用于等于,if 范围更广
- switch 中不使用 break,可一直执行
- case