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)
链接:https://leetcode-cn.com/problems/iterator-for-combination
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class CombinationIterator:

    def __init__(self, characters: str, combinationLength: int):
        self.cha = characters
        # 建立列表用来保存各种字母组合在characters中的索引
        self.com = [x for x in range(combinationLength)]
        self.finished = False

    def next(self) -> str:
    	# 当前排列方式所对应的字符串
        ans = "".join([self.cha[i] for i in self.com])
        # 此值用来判断是否为最后一种字母组合
        x = -1
        # 判断当前字母组合是否为最后一种
        for i in range(len(self.com) - 1, -1, -1):
            if self.com[i] != len(self.cha) - len(self.com) + i:
                x = i
                break
        if x == -1:
            self.finished = True
        else:
        	# 切换为下种字母组合
            self.com[x] += 1
            # 若某索引进行更改,则其后面的索引都必须增大,否则会出现冲突
            for j in range(x + 1, len(self.com)):
                self.com[j] = self.com[j - 1] + 1
        return ans

    def hasNext(self) -> bool:
    	# 若为最后一种排列方式则不存在下一个字母组合
        return not self.finished  


# Your CombinationIterator object will be instantiated and called as such:
# obj = CombinationIterator(characters, combinationLength)
# param_1 = obj.next()
# param_2 = obj.hasNext()
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值