链表在指定时不返回任何内容

Question

我正在创建一个单链表，它返回给定范围内的奇数，而不是返回 1、3、5 等，而是返回 1、无、3、无、5 等。我想做以便它停止返回 None 并且只返回奇数。

class Odds:
    def __init__(self,end):
        self.__start = 1
        self.__end = end

    def __iter__(self):
        return OddsIterator(self.__end)

class OddsIterator:
    def __init__(self,finish): 
        self.__current = 0
        self.__step = 1
        self.__end = finish

    def __next__(self):
        x = None
        if self.__current > self.__end:
            raise StopIteration 
        else:
            self.__current += self.__step 
            if (self.__current - self.__step + 1) % 2 != 0:
                x = self.__current - self.__step + 1
        if x != None:
            return x 

Answer 1

返回None 的原因是在没有return 语句的情况下有可能到达__next__ 的末尾，在这种情况下，任何Python 函数都会返回None。这里的解决方案是在初始化的时候设置self.__step = 2和self.__current = 1。您也可以删除if x != None:，因为它什么都不做。为了适应self.__current 的新定义，我在下面做了一些其他的小改动。目前下面的代码不会包含最终值，如果你想把self.__current >= self.__end改为self.__current > self.__end。

class Odds:
    def __init__(self,end):
        self.__start = 1
        self.__end = end

    def __iter__(self):
        return OddsIterator(self.__end)

class OddsIterator:
    def __init__(self,finish):
        self.__current = 1
        self.__step = 2
        self.__end = finish

    def __next__(self):
        x = None
        if self.__current >= self.__end:
            raise StopIteration
        else:
            return_value = self.__current
            self.__current += self.__step
            return return_value

odds = Odds(21)
print(list(odds))