Python-68：多米诺骨牌均衡状态

问题描述

小S玩起了多米诺骨牌，他排列了一行骨牌，并可能将某些骨牌向左或向右推倒。随着骨牌连锁反应的进行，一些骨牌可能因为左右两侧受力平衡而保持竖立。现在小S想要知道在所有动作完成后，哪些骨牌保持竖立。

给定一个表示骨牌初始状态的字符串，其中：

• "L" 表示该位置的骨牌将向左倒。
• "R" 表示该位置的骨牌将向右倒。
• "." 表示该位置的骨牌初始时保持竖立。

模拟整个骨牌倒下的过程，求出最终仍然保持竖立的骨牌的数目和位置。

代码

def solution(num, data):

    # Please write your code here

    assert num == len(data)

    cnt = [0] * 3010

    ans = 0

    for i in range(num):

        if data[i] == 'L':

            j, k = 1, i - 1

            while k >= 0 and data[k] == '.':

                cnt[k] -= j

                k -= 1

                j += 1

        elif data[i] == 'R':

            j, k = 1, i + 1

            while k < num and data[k] == '.':

                cnt[k] += j

                k += 1

                j += 1

    for i in range(num):

        if cnt[i] == 0 and data[i] == '.':

            ans += 1

    s = str(ans)

    if ans == 0:

        return s

    num_count = 0

    result = []

    for i in range(num):

        if cnt[i] == 0 and data[i] == '.':

            num_count += 1

            if num_count != ans:

                result.append(str(i + 1))

            else:

                result.append(str(i + 1))

    return s + ":" + ",".join(result)

if __name__ == "__main__":

    #  You can add more test cases here

    print(solution(14, ".L.R...LR..L..") == "4:3,6,13,14" )

    print(solution(5, "R....") == "0" )

    print(solution(1, ".") == "1:1" )