Вы можете найти все подстроки текущей руки и отфильтровать результаты, чтобы найти те, которые отсортированы с шагом 1
:
def stringify_result(f):
def wrapper(_d):
cards = {10:'T', 11:'J', 12:'Q', 13:'K'}
return list(map(lambda x:cards.get(x, str(x)), f(_d)))
return wrapper
@stringify_result
def has_straight(d):
cards = {'J': 11, 'K': 13, 'T': 10, 'Q': 12}
subs = list(filter(None, [d[b:i] for i in range(len(d)+1) for b in range(len(d)+1)]))
possibilities = list(filter(lambda x:all(x[i+1] - x[i] == 1 for i in range(len(x)-1)), [[int(cards.get(b, b)) for b in i] for i in subs]))
return [] if not possibilities else max(possibilities, key=len)
straight = has_straight(['6', '8', '8', '9', 'T'])
score = len(straight)
Вывод:
['8', '9', 'T']
3
Изменить: для учета нескольких прогонов, вы можете использовать itertools.groupby
:
import itertools
def has_straight(d):
cards = {'J': 11, 'K': 13, 'T': 10, 'Q': 12}
mutated_cards = list(map(lambda x:int(cards.get(x, x)), d))
_grouped = [list(b) for _, b in itertools.groupby(mutated_cards)]
subs = list(filter(None, [_grouped[b:i] for i in range(len(_grouped)+1) for b in range(len(_grouped)+1)]))
final_filtered = list(filter(lambda x:all(x[i+1][0] - x[i][0] == 1 for i in range(len(x)-1)), subs))
new_subs = [] if not final_filtered else max(final_filtered, key=lambda x:sum(len(i) for i in x))
return sum(len(i) for i in new_subs if len(i) > 1), list(map(lambda x:x[0], new_subs))
print(has_straight(['6', '8', '8', '9', 'T']))
print(has_straight(['4', '4', '5', '5', '6']))
Выход:
(2, [8, 9, 10])
(4, [4, 5, 6])