How to Merge Two List/Iterators in Java?

Time：2020-09-09 14:04:20
Class：Weblog
Read：37

We have known the approach to merge two linked list in C++. We also know how to apply the merge sort algorithms to merge K sorted list. If you are given two Iterators in Java, how are you supposed to merge them into one list if both iterators are sorted?

The Java’s iterator has two important methods you can use: hasNext() and next(). The hasNext() returns a boolean telling if this iterator reaches the end. And the next() will advance to next position and returns the value of it. It will throw exception if the iterator is beyond the end (EOF).

Merging Two Sorted Iterators in Java

Remember, everytime we call next() we have to put a safety check hasNext(). Then the logic is pretty simple: we first deal with the merging when both iterators are not NULL.

Then, we have to follow the remaining iterator and copy over the values.

package helloacm.com
import lombok.var;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
 
public class Main {
    public static List<Integer> merge(Iterator<Integer> it1, Iterator<Integer> it2) {
        List<Integer> res = new ArrayList<>();
        var v1 = it1.hasNext() ? it1.next() : null;
        var v2 = it2.hasNext() ? it2.next() : null;
        while (v1 != null && v2 != null) {
            if (v1 < v2) { // pick the smaller one
                res.add(v1);
                v1 = it1.hasNext() ? it1.next() : null;
            } else {
                res.add(v2);
                v2 = it2.hasNext() ? it2.next() : null;
            }
        }
        while (v1 != null) {
            res.add(v1);
            v1 = it1.hasNext() ? it1.next() : null;
        }
        while (v2 != null) {
            res.add(v2);
            v2 = it2.hasNext() ? it2.next() : null;
        }
        return res;
    }
 
    public static void main(String[] args) {
        var l1 = Arrays.asList(1, 3, 5, 7, 9, 11, 15, 17);
        var l2 = Arrays.asList(2, 4, 6, 8, 10, 12, 16, 18);
        var sorted = merge(l1.iterator(), l2.iterator());
        for (var num: sorted) {
            System.out.println(num);
        }
    }
}

package helloacm.com
import lombok.var;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;

public class Main {
    public static List<Integer> merge(Iterator<Integer> it1, Iterator<Integer> it2) {
        List<Integer> res = new ArrayList<>();
        var v1 = it1.hasNext() ? it1.next() : null;
        var v2 = it2.hasNext() ? it2.next() : null;
        while (v1 != null && v2 != null) {
            if (v1 < v2) { // pick the smaller one
                res.add(v1);
                v1 = it1.hasNext() ? it1.next() : null;
            } else {
                res.add(v2);
                v2 = it2.hasNext() ? it2.next() : null;
            }
        }
        while (v1 != null) {
            res.add(v1);
            v1 = it1.hasNext() ? it1.next() : null;
        }
        while (v2 != null) {
            res.add(v2);
            v2 = it2.hasNext() ? it2.next() : null;
        }
        return res;
    }

    public static void main(String[] args) {
        var l1 = Arrays.asList(1, 3, 5, 7, 9, 11, 15, 17);
        var l2 = Arrays.asList(2, 4, 6, 8, 10, 12, 16, 18);
        var sorted = merge(l1.iterator(), l2.iterator());
        for (var num: sorted) {
            System.out.println(num);
        }
    }
}

Unlike linked lists, for iterators, you have to iterate and copy over the values. With linked list, you can just re-point/re-connect the head with O(1) constant time.

The runtime complexity is O(M+N) where M and N are the number of the elements in each iterator respectively. The space requirement is also O(M+N).

–EOF (The Ultimate Computing & Technology Blog) —

Recommend：
Summits set epoch-making milestone in history of China-Arab ties
In the face of COVID-19 pandemic, China and Arab countries have
15 Macao residents qualify as candidates for deputies to nationa
Study finds genetic solution to pre-harvest sprouting in rice, w
Bodybuilders dying as coaches, judges encourage extreme measures
Malta's Marsaskala, China's Dujiangyan sign sister city agreemen
U.S. mortgage applications continue slide amid surging interest
Russian, UAE presidents discuss bilateral cooperation over phone
Hate crimes in U.S. Los Angeles County rise to highest level sin
Chinese mainland reports 4,031 new local confirmed COVID-19 case

Comment list

Comment add: Name：　Check：