package cn.leetcode.demo;

import java.util.Hashtable;
import java.util.Scanner;

/*
 *看一个实际需求，google公司的一个上机题: 
有一个公司,当有新的员工来报道时,要求将该员工的信息加入(id,性别,年龄,住址..),
当输入该员工的id时,要求查找到该员工的 所有信息.

要求: 不使用数据库,尽量节省内存,速度越快越好=>哈希表(散列)
 * @author aparke
 *
 */
public class HashTabDemo {
	@SuppressWarnings("resource")
	public static void main(String[] args) {
		//测试
		//创建一个哈希表
		HashTab hashTab = new HashTab(4);
		
		//创建一个菜单，完成测试
		String key="";
		Scanner scanner = new Scanner(System.in);
		while (true) {

			System.out.println("add：添加雇员");
			System.out.println("list: 显示雇员");
			System.out.println("find: 查找雇员");
			System.out.println("exit: 退出系统");
			key=scanner.next();
			switch (key) {
			case "add":
				System.out.println("添加一个雇员");
				System.out.println("输入id");
				int id =scanner.nextInt();
				System.out.println("输入名字");
				String name = scanner.next();
				//创建雇员
				Emp emp=new Emp(id, name);
				hashTab.add(emp);
				break;
				
			case "list":
				hashTab.list();
				break;	
				
			case "find":
				hashTab.list();
				System.out.println("输入要查找的id");
				id =scanner.nextInt();
				hashTab.findEmpById(id);
				break;	
			
			case "exit":
				System.exit(0);

			default:
				break;
			}
		}
	}

}

//1.创建雇员类
class Emp {
	public int id;
	public String name;
	public Emp next;// 指向下一个节点的引用，默认为空
	// 创建一个构造器

	public Emp(int id, String name) {
		super();
		this.id = id;
		this.name = name;

	}
}

//2.创建EmpLinkedList ,表示链表
class EmpLinkedList {
	//// 头指针，执行第一个Emp,因此我们这个链表的head 是直接指向第一个Emp
	private Emp head;// 默认null

	// 添加雇员到链表
	/*
	 * 说明： 1).假定，当添加雇员时，id为自增长，即id分配后总是从小到大 2).因此，我们将该雇员添加到链表最后即可
	 */
	public void add(Emp emp) {
		// 如果添加第一个雇员,head 头指针时不能动
		if (head == null) {
			head = emp;
			return;
		}
		// 如果不是第一个雇员，则使用一个辅助的指针，帮助定位到最后
		Emp curEmp = head;
		while (true) {
			if (curEmp.next == null) {// 表示到达了链表的最后
				break;
			}
			// 否则后移，
			curEmp.next = curEmp;
		}
		// 将curEmp加入到链表
		curEmp.next = emp;
	}

	// 遍历链表的雇员信息
	public void list(int no) {
		if (head == null) {
			System.out.println("第 " + (no + 1) + " 链表为空");
			return;
		}
		System.out.print("第 " + (no + 1) + " 链表的信息为");
		Emp curEmp = head;// 辅助指针
		while (true) {
			System.out.printf(" => id=%d name=%s\t", curEmp.id, curEmp.name);
			if (curEmp.next == null) {
				break;
			}
			curEmp = curEmp.next;// 后移
		}
		System.out.println();
	}

	// 根据id查找雇员
	// 如果找到就返回emp，如果未找到就返回null
	public Emp findEmpByid(int id) {
		// 判断链表是否为空
		if (head == null) {
			System.out.println("链表为空");
			return null;
		}
		// 辅助指针
		Emp curEmp = head;
		while (true) {
			if (curEmp.id==id) {
				break;//找到返回，此时curEmp就指向要查找的雇员，然后退出
			}
			//否者继续判断，如果curEmp.next==null的，则说明遍历该链表没有找到该雇员
			if (curEmp.next==null) {
				curEmp=null;
				break;
			}
			curEmp=curEmp.next;
		}

		return curEmp;
	}
}

//3.创建HashTab 管理多条链表
class HashTab {
	private EmpLinkedList[] empLinkedListArray; // 定义一个数组存放链表
	private int size; // 表示有多少条链表

	// 构造器，初始化链表
	public HashTab(int size) {
		this.size = size;

		empLinkedListArray = new EmpLinkedList[size];

		// 初始化多个链表，在声明对象数组后 ，必须对每个数组成员进行实例化话 才能直接使用，否则报空指针异常！
		for (int i = 0; i < size; i++) {
			empLinkedListArray[i] = new EmpLinkedList();
		}
	}

	// 编写散列函数，使用一个简单的取模法
	public int hashFun(int id) {
		return id % size;
	}

	// 添加雇员
	public void add(Emp emp) {
		// 根据员工的id,得到该员工应当添加到哪个链表中,通过取模函数使其添加到那条链表中
		int emLinkedListNo = hashFun(emp.id);
		// 将emp 添加到对应的链表中
		empLinkedListArray[emLinkedListNo].add(emp);
	}

	// 遍历所有的链表,遍历hashtab
	public void list() {
		for (int i = 0; i < size; i++) {
			empLinkedListArray[i].list(i);
			;
		}
	}

	// 根据输入的id，查找雇员
	public void findEmpById(int id) {
		// 使用散列函数确定到那条链表查找
		int empLinkedListNo = hashFun(id);
		Emp emp=empLinkedListArray[empLinkedListNo].findEmpByid(id);
		if (emp!=null) {
			System.out.printf("在第%d条链表中找到 雇员 id = %d\n", (empLinkedListNo + 1), id);
		}else {
			System.out.println("在哈希表中，没有找到该雇员~");
		}

	}
}