merge k sorted list with iterator

最新推荐文章于 2026-06-14 14:48:46 发布

原创最新推荐文章于 2026-06-14 14:48:46 发布 · 804 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

strStr 同时被 2 个专栏收录

360 篇文章

订阅专栏

队列

63 篇文章

订阅专栏

本文介绍了一种实现合并K个已排序整数列表的方法，通过使用Java中的优先队列来按顺序输出所有元素。具体实现了MergeKListIterator类，构造函数接收多个列表并初始化优先队列，hasNext()方法检查是否还有更多元素，next()方法返回下一个最小元素。

Given k sorted lists of O(n) integers each, implement an iterator that will yield all elements in sorted order。

package list;

import java.util.*;

class Wrapper {
	public ArrayList<Integer> list;
	public int index;

	public Wrapper(ArrayList<Integer> list, int index) {
		this.list = list;
		this.index = index;
	}
}

public class MergeKListIterator {
	PriorityQueue<Wrapper> queue;

	public MergeKListIterator(ArrayList<ArrayList<Integer>> lists) {
		queue = new PriorityQueue<Wrapper>(10, new Comparator<Wrapper>() {
			@Override
			public int compare(Wrapper w1, Wrapper w2) {
				return w1.list.get(w1.index) - w2.list.get(w2.index);
			}
		});

		for (ArrayList<Integer> list : lists) {
			Wrapper w = new Wrapper(list, 0);
			queue.offer(w);
		}
	}

	public boolean hasNext() {
		if (!queue.isEmpty())
			return true;

		return false;
	}

	public Integer next() {

		Wrapper w = queue.poll();

		ArrayList<Integer> list = w.list;
		int index = w.index;

		if (index < list.size() - 1) {
			w.index = index + 1;
			queue.offer(w);
		}

		return list.get(index);
	}

	public static void main(String[] args) {

		ArrayList<Integer> l1 = new ArrayList<Integer>();
		ArrayList<Integer> l2 = new ArrayList<Integer>();
		ArrayList<Integer> l3 = new ArrayList<Integer>();

		l1.add(2);
		l1.add(4);
		l1.add(6);
		l2.add(1);
		l2.add(3);
		l3.add(5);

		ArrayList<ArrayList<Integer>> lists = new ArrayList<ArrayList<Integer>>();
		lists.add(l1);
		lists.add(l2);
		lists.add(l3);

		MergeKListIterator mk = new MergeKListIterator(lists);
		while (mk.hasNext()) {
			System.out.println(mk.next());
		}
	}

}