题目概述
解题思路
题意是:给定一个盒子(用二维矩阵表示),盒子里面每个格子有三种情况:空/有可以移动的石头/有不能移动的障碍。现在把水平放置的盒子竖直摆过来,问盒子里面会发生怎样的变化。
这道题的做法是:单独分析每一行的变化。对于每一行来说,我们从右往左分析逐个格子(表示当前指针所指位置),维护一个变量,记录当前行最靠右边的障碍的位置
:
- 如果格子里是障碍,则更新
;
- 如果格子里是石头,则将石头移动到
的位置。
方法性能
时间复杂度是O(MN):
示例代码
class Solution {
public:
vector<vector<char>> rotateTheBox(vector<vector<char>>& box) {
int R_len = box.size(),
C_len = box[0].size();
int cur_pos;
for(int ri = 0; ri < R_len;++ri)
{
cur_pos = C_len;
for(int ci = C_len - 1; ci >= 0;--ci)
{
//meet wall
if(box[ri][ci] == '*')
cur_pos = ci;
else if(box[ri][ci] == '#') //meet stone
{
//move stone
if(cur_pos - ci > 1)
{
cur_pos--;
box[ri][cur_pos] = '#';
box[ri][ci] = '.';
}
else
{
cur_pos = ci;
}
}
}
}
vector<vector<char>> res(C_len, vector<char>(R_len));
for(int ri = 0; ri < R_len; ++ri)
{
for(int ci = 0; ci < C_len;++ci)
{
res[ci][R_len - ri - 1] = box[ri][ci];
}
}
return res;
}
};