Leetcode 1286：字母组合迭代器（超详细的解法！！！）

请你设计一个迭代器类，包括以下内容：

• 一个构造函数，输入参数包括：一个 有序且字符唯一 的字符串 characters（该字符串只包含小写英文字母）和一个数字 combinationLength 。
• 函数 next() ，按 字典序 返回长度为 combinationLength 的下一个字母组合。
• 函数 hasNext() ，只有存在长度为 combinationLength 的下一个字母组合时，才返回 True；否则，返回 False。

示例：

CombinationIterator iterator = new CombinationIterator("abc", 2); // 创建迭代器 iterator

iterator.next(); // 返回 "ab"
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 "ac"
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 "bc"
iterator.hasNext(); // 返回 false

提示：

• 1 <= combinationLength <= characters.length <= 15
• 每组测试数据最多包含 10^4 次函数调用。
• 题目保证每次调用函数 next 时都存在下一个字母组合。

解题思路

这个问题实际上就是Leetcode 77：组合（最详细的解法！！！），所以我们可以通过之前的问题计算出所有的组合，然后通过队列将所有的组合存储，每次调用next的时候将队首元素弹出即可。

class CombinationIterator:
    def __init__(self, ch: str, l: int):
        self.res = []
        
        def combine(s, cur):
            if len(cur) == l:
                self.res.append(cur)
                return
            for i in range(s, len(ch) - (l - len(cur)) + 1):
                combine(i + 1, cur + ch[i])
                
        combine(0, "")
        
    def next(self) -> str:
        return self.res.pop(0)
        
    def hasNext(self) -> bool:
        return len(self.res) != 0

python中有相关函数的包，所以我们也可以直接调用。

class CombinationIterator:
    def __init__(self, ch: str, l: int):
        self.data = itertools.combinations(ch, l)
        self.last = ch[-l:]
        self.res = None

    def next(self) -> str:
        self.res = "".join(next(self.data))
        return self.res

    def hasNext(self) -> bool:
        return self.last != self.res

更加pythonic的写法

class CombinationIterator:
    def __init__(self, ch: str, l: int):
        data = list(map("".join, itertools.combinations(ch, l)))[::-1]
        self.next = data.pop
        self.hasNext = lambda : bool(data)

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题，希望大家指出！！！