README.md

Linked List with Iterator

An example of flattening a linked list with generators.

Usage

const LinkedList = require('./linked-list');

const list = new LinkedList();

list.push(10);
list.push('foo');
list.push('bar');
list.push(42);

console.log([...list]);
// [ 10, 'foo', 'bar', 42 ]

linked-list.js

class Node {
  constructor(value, next = null) {
    this.value = value;
    this.next = next;
  }

  *[Symbol.iterator]() {
    yield this.value;
    if (this.next !== null) { yield * this.next; }
  }
}

module.exports = class LinkedList {
  constructor() { this.head = null; }

  push(value) {
    this.head = new Node(value, this.head);
  }

  peek() {
    if (this.head === null) { return null; }
    return this.head.value;
  }

  pop() {
    const old = this.head;
    this.head = this.head.next;
    return old.value;
  }

  *[Symbol.iterator]() {
    if (this.head !== null) { yield * this.head; }
  }
}