推倒多米诺骨牌
問題描述
有 N 張多米諾骨牌,將每張多米諾骨牌垂直豎立。在開始時,同時把一些多米諾骨牌向左或向右推(初始狀態)。以字符串的形式給定多米諾骨牌的初始狀態,在給定字符串中:
. 代表該位置的多米諾骨牌保持靜止。
L 代表將該位置的多米諾骨牌向左倒。
R 代表將該位置的多米諾骨牌向右倒。
每過一秒,倒向左邊的多米諾骨牌會推動其左側相鄰的多米諾骨牌。同樣地,倒向右邊的多米諾骨牌也會推動豎立在其右側的相鄰多米諾骨牌。如果同時有多米諾骨牌落在一張垂直豎立的多米諾骨牌的兩邊,由于受力平衡, 該骨牌仍然保持不變。基于以上考慮,輸出對給定初始狀態的多米諾骨牌穩定后的最終狀態。
測試樣例
輸入: '..R.....L..R.' 輸出: '..RRR.LLL..RR'內容首發于微信公眾號IT信息教室,如果您想學習更多AI相關的技能,歡迎搜索關注或微信掃描下方二維碼關注~~
參考代碼
# O(n) time, O(n) space, where n is the number of dominoes. class Solution():def pushDominoes(self, dominoes):N = len(dominoes)force = [0] * N# Populate force from left to right.currentForce = 0for idx in range(N):if dominoes[idx] == 'R':currentForce = Nelif dominoes[idx] == 'L':currentForce = 0else:currentForce = max(currentForce - 1, 0)force[idx] += currentForce# Populate force from right to left.currentForce = 0for idx in reversed(range(N)):if dominoes[idx] == 'L':currentForce = -Nelif dominoes[idx] == 'R':currentForce = 0else:currentForce = min(currentForce + 1, 0)force[idx] += currentForcereturn force# Convert list to string.def getFinalState(self, dominoes):finalState = self.pushDominoes(dominoes)result = ''for currentForce in finalState:if currentForce == 0:result += '.'elif currentForce > 0:result += 'R'elif currentForce < 0:result += 'L'return result# Test Program dominoes = '..R.....L..R.' print(Solution().getFinalState(dominoes)) #..RRR.LLL..RR總結