python - 从给定的纸牌中直接检测并返回

你得到了一副牌。这套衣服不重要。可以重复。卡片用数字1到13表示。
目标是在给定的列表中找到所有可能的直线直卡定义为5张连续卡：1, 2, 3, 4, 5是直卡，10, 11, 12, 13, 1和5,6,7,8,9,10不是。我为此编写了代码，但我的代码并不适用于所有情况：

def is_straight(array):
    n = len(array)

    if n % 5 != 0:
        return False

    array.sort()
    prev = array[0]
    if n == 5:
        output = []
        i = 1
        while i < n:
            if array[i] - prev == 1:
                output.append(array[i - 1])
                prev = array[i]
                i += 1
            else:
                return False
        output.append(prev)
        return output

    else:
        l = int(n / 5)
        st_list = [[] for i in range(l)]
        set_track = 0
        for i in range(1, n):
            if array[i] - prev == 1:
                for k in range(0, len(st_list)):
                    if prev not in st_list[k]:
                        st_list[k].append(array[i-1])
                        set_track = k
                        break
            elif array[i] - prev == 0:
                for k in range(0, len(st_list)):
                    if prev not in st_list[k]:
                        st_list[k].append(array[i-1])
                        set_track = k
                        break
            elif array[i] - prev > 1:
                del st_list[set_track]
                l -= 1
            prev = array[i]
        st_list[l-1].add(prev)
        return st_list


print(is_straight([1,2,3,4,4,6,7,8,9,10, 1,2,3,4,4,6,7,8,9,10]))

请告诉我如何修复此代码。上面的输入不起作用。

最佳答案

你的解决方案有几个问题我相信你把问题弄得比需要的复杂多了如果我们想得对的话，我们可以在O(N)中解决这个问题。
1）由于我们有有界输入，而且我们不关心重复的卡，所以我们可以使用一个标志数组如果我们有卡就把位置设为真，如果没有卡就设为假
2）由于我们需要五张连续的卡片，所以如果我们巧妙地使用模数并跟踪连续的卡片，我们只需要循环遍历一次数组把这一切放在一起，我们可以得到

def sol(myHand):
    hand = [False for _ in range(13)]
    for card in myHand:
        hand[card-1] = True

    for i in range(len(hand)):
        staightCounter = 0
        while hand[i % 13] and staightCounter < 5:
            i += 1
            staightCounter += 1
        if staightCounter >= 5:
            print([ x % 13 + 1 for x in range(i-5,i)])

关于python - 从给定的纸牌中直接检测并返回，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/51508170/