[程序员代码面试指南]第9章-一种消息接收并打印的结构（链表）

题意

数据流读入数字(保证>=1)，i最初=0，当i+1出现时，输出i+1及其后连续的输入过的数字，否则什么也不输出。
例子：
输入：1 输出：1
输入：3 输出：
输入：4 输出：
输入：6 输出：
输入：2 输出：2 3 4

题解

用链表+headMap+tailMap存。
总时间复杂度O(n).

代码

import java.util.HashMap;
import java.util.Scanner;

class Node{
	int val;
	Node next;
	public Node(int val) {
		this.val=val;
	}
}

class Message{
	HashMap<Integer,Node> headMap=new HashMap();
	HashMap<Integer,Node> tailMap=new HashMap();
	int lastPrint=0;
	
	public void receive(int num) {
		Node node=new Node(num);
		headMap.put(num, node);
		tailMap.put(num,node);
		
		if(headMap.containsKey(num+1)) {
			node.next=headMap.get(num+1);
			headMap.remove(num+1);
			headMap.put(num,node);
		}
		if(tailMap.containsKey(num-1)) {
			tailMap.get(num-1).next=node;
			tailMap.remove(num-1);
			tailMap.put(num,node);
		}
		if(headMap.containsKey(lastPrint+1)) {
			print();
		}
	}
	
	private void print() {//
		Node node=headMap.get(lastPrint+1);
		headMap.remove(lastPrint+1);
		while(node.next!=null) {
			System.out.println(node.val);
			++lastPrint;
			node=node.next;
		}
		System.out.println(node.val);
		++lastPrint;
		tailMap.remove(node.val);
	}
}

public class Main {
	public static void main(String args[]) {
		Scanner in=new Scanner(System.in);
		Message mes=new Message();
		while(in.hasNext()) {
			int num=in.nextInt();
			mes.receive(num);
		}
	}
}