1 import itertools 2 class CombinationIterator: 3 4 def __init__(self, characters: str, combinationLength: int): 5 self.lists = list(itertools.combinations(characters,combinationLength)) 6 self.index = 0 7 8 def next(self) -> str: 9 tp = self.lists[self.index] 10 self.index += 1 11 return ''.join(tp) 12 13 def hasNext(self) -> bool: 14 return self.index < len(self.lists)
直接调用itertools内置函数,快速生成符合条件的组合。
1286. Iterator for Combination