leetcode - 838. 推多米诺

leetcode - 838. 推多米诺

题目

代码

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

//方法一：广度优先搜索 
string pushDominoes(string dominoes) {
	int n = dominoes.size();
	queue<int> q; //模拟搜索的顺序
	vector<int> time(n, -1);  //记录骨牌翻到或者不翻到的时间 
	vector<string> force(n);  //记录骨牌 
	for(int i = 0; i < n; i++){
		if(dominoes[i] != '.'){
			q.push(i);
			time[i] = 0;
			force[i].push_back(dominoes[i]); 
		}
	}
	string res(n, '.');
	while(!q.empty()){
		int i = q.front();
		q.pop();
		if(force[i].size() == 1){
			char f = force[i][0];
			res[i] = f;
			int ni = (f == 'L') ? (i-1) : (i+1);
			if(ni >=0 && ni < n){
				int t = time[i];
				if(time[ni] == -1){
					q.push(ni);
					time[ni] = t+1;
					force[ni].push_back(f);
				}else if(time[ni] == t+1){
					force[ni].push_back(f);
				}
			}
		}
	}
	return res;
}

/*
方法二：模拟
确定所有连续的没有被推动的骨牌，根据这段骨牌的两边骨牌的推倒方向决定这段骨牌的最终状态
1）如果两边骨牌同向，那么这段连续竖立的骨牌会倒向同一方向（LL,RR）
2）如果两边骨牌相对，那么这段连续竖立的骨牌会向中间倒（RL）
3）如果两边骨牌相反，那么这段连续竖立的骨牌会保持中立（LR） 
*/ 
string pushDominoes2(string dominoes) {
        int n = dominoes.size(), i = 0;
        char left = 'L';
        while (i < n) {
            int j = i;
            while (j < n && dominoes[j] == '.') { // 找到一段连续的没有被推动的骨牌
                j++;
            }
            char right = j < n ? dominoes[j] : 'R';
            if (left == right) { // 方向相同，那么这些竖立骨牌也会倒向同一方向
                while (i < j) {
                    dominoes[i++] = right;
                }
            } else if (left == 'R' && right == 'L') { // 方向相对，那么就从两侧向中间倒
                int k = j - 1;
                while (i < k) {
                    dominoes[i++] = 'R';
                    dominoes[k--] = 'L';
                }
            }
            left = right;
            i = j + 1;
        }
        return dominoes;
}

int main(){
	string res;
	string dominoes;
	cin>>dominoes;
	res = pushDominoes(dominoes);
	cout<<res;
	return 0;
}