DIY: Flatten Nested List Iterator

Solve the interview question "Flatten Nested List Iterator" in this lesson.

Problem statement

You will be given a nested list of integers named nested_list. Each element will either be an integer or a list whose elements may also be integers or other lists. Your task will be to implement an iterator to flatten the nested list.

You will have to implement the NestedIterator module. This class has the following functions:

  • init(nested_list), which will initialize the iterator with the list nested_list.
  • next(obj), which will return the next integer in the nested list.
  • has_next(obj), which will return true if there are still some integers in the nested list and, otherwise, will return false.

You will be given a class definition for NestedInteger and some helper functions. Make sure to look at them in the NestedIntegers.ex file, in the exercise given below.

Level up your interview prep. Join Educative to access 80+ hands-on prep courses.