Default Arguments

  • + 0 comments

    class EvenStream: def init(self): self.current = 0

    def get_next(self):
        next_value = self.current
        self.current += 2
        return next_value
    

    class OddStream: def init(self): self.current = 1

    def get_next(self):
        next_value = self.current
        self.current += 2
        return next_value
    

    def print_from_stream(n, stream=None): if stream is None: stream = EvenStream() for _ in range(n): print(stream.get_next())

    if name == "main": queries = int(input()) for _ in range(queries): stream_name, n = input().split() n = int(n) if stream_name == "even": print_from_stream(n) elif stream_name == "odd": print_from_stream(n, OddStream())