Implement a Lazy Tee Iterator in Python

Question

This question asks the candidate to implement a Lazy Tee Iterator in Python, which allows for splitting an iterable into two separate iterators that can be consumed independently.

Codemia · Accepted Answer

To implement a Lazy Tee Iterator, I will create a class that takes an iterable as input and yields values from that iterable to two separate iterators. This approach ensures that both iterators can be consumed independently without duplicating the data in memory.

Here is a basic implementation of the Lazy Tee Iterator in Python:

```python
class LazyTee:
    def __init__(self, iterable):
        self.iterable = iterable
        self.iter1 = []
        self.iter2 = []

def __iter__(self):
        for item in self.iterable:
            self.iter1.append(item)
            yield item

def tee(self):
        return iter(self.iter1), iter(self.iter2)
```

In this implementation, the `__iter__` method yields items from the original iterable while storing them in two separate lists for the two iterators.

The time complexity of this implementation is O(n), where n is the number of items in the iterable, since we iterate through the entire input once. The space complexity is also O(n) due to the storage of items in the two lists.

This implementation is straightforward but may not be memory efficient for very large datasets since it stores all items in memory. An alternative approach could involve using a generator to yield items to both iterators without storing them, but this would require additional logic to manage the state of each iterator.

It is important to handle cases where the input iterable is empty or not an iterable type. I would add type checks and raise appropriate exceptions to ensure robustness. For example:

```python
if not hasattr(iterable, '__iter__'):
    raise TypeError('Input must be an iterable.')
```

Implement a Lazy Tee Iterator in Python

Google

What the Interviewer Expects

Key Topics to Cover

How to Approach This

Possible Follow-up Questions

Sharpen Your Skills on Codemia

Sample Answer

Approach

Implementation

Submit Your Answer

Google Software Engineer Interview Guide

Related Questions

Design a scalable Search System

Design a Video Streaming for Rippling

Design a Load Balancing Service

Design a Image Processing Service

Build a high-throughput Image Processing Pipeline