题目地址:
https://leetcode.com/problems/push-dominoes/
给定
n
n
n个多米诺骨牌,每个骨牌可能会施加一个力,力的方向只有左和右两种,并且力是同时施加的。施加力的情况会以一个字符串的形式给出,'R'
和'L'
代表向右和向左,'.'
代表未施加力。在施加之后,每个骨牌都会按照力的方向倒下,除非遇到了一个相反方向倒下的骨牌。如果某个骨牌同时在左右被两个骨牌撞到,则其不会倒。问最后每个骨牌倒下的方向。
首先处理一下形如"R...L"
的情形,必然会从两边依次填上'R'
和'L'
直到中间。然后处理向右倒的情形,如果某个骨牌左边的骨牌是向右倒的,那么它自己也会向右倒,除非它的右边是'L'
(由于"R...L"
的情形已经处理过了,这里右边是'L'
仅仅指形如"R.L"
的情况);接下来再处理一下向左倒的情况就行了。代码如下:
public class Solution {
public String pushDominoes(String s) {
int n = s.length();
char[] chs = s.toCharArray();
// R ... L
for (int i = 0, pos = -1; i < n; i++) {
char ch = s.charAt(i);
if (ch == '.') {
continue;
}
if (ch == 'R') {
pos = i;
} else {
if (pos != -1) {
// 从两边向中间填
for (int l = pos + 1, r = i - 1; l < r; l++, r--) {
chs[l] = 'R';
chs[r] = 'L';
}
// 处理完一个要重置pos
pos = -1;
}
}
}
for (int i = 1; i < n; i++) {
if (chs[i] == '.' && chs[i - 1] == 'R' && (i == n - 1 || chs[i + 1] != 'L')) {
chs[i] = 'R';
}
}
for (int i = n - 2; i >= 0; i--) {
if (chs[i] == '.' && chs[i + 1] == 'L' && (i == 0 || chs[i - 1] != 'R')) {
chs[i] = 'L';
}
}
return new String(chs);
}
}
时空复杂度 O ( n ) O(n) O(n)。