哈希表的底层是一个数组,数组中的每个元素对应了一个链表。
package com.atguigu.hashtab;
import java.util.Scanner;
public class HashTabDemo {
public static void main(String[] args) {
//创建一个哈希表
HashTab hashTab=new HashTab(7);
//写一个简单的菜单
String key="";
Scanner scanner=new Scanner(System.in);
while(true) {
System.out.println("add:添加雇员");
System.out.println("list:显示雇员");
System.out.println("exit:退出系统");
System.out.println("find:查找雇员");
System.out.println("del:删除雇员");
key=scanner.next();
switch(key) {
case "add":
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":
System.out.println("请输入要查找的id");
id=scanner.nextInt();
hashTab.findEmp(id);
break;
case "del":
System.out.println("请输入要删除的id");
id=scanner.nextInt();
hashTab.deleteEmp(id);
break;
case "exit":
scanner.close();
System.exit(0);
default:
break;
}
}
}
}
//创建HashTab 管理多条链表
class HashTab{
private EmpLinkedList[] empLinkedListArray;
private int size;//表示共有多少条链表
//构造器
public HashTab(int size) {
//初始化empLinkedListArray
this.size=size;
empLinkedListArray=new EmpLinkedList[size];
//不要忘记初始化每个链表
for(int i=0;i<size;++i) {
empLinkedListArray[i]=new EmpLinkedList();
}
}
//添加雇员
public void add(Emp emp) {
//根据员工的ID得到该员工应当添加到哪条链表
int empLinkedListNo=hashFun(emp.id);
//将emp添加到对应的链表中
empLinkedListArray[empLinkedListNo].add(emp);
}
//遍历所有链表,即遍历hashTab
public void list() {
for(int i=0;i<size;i++) {
empLinkedListArray[i].list(i+1);
}
}
//根据输入的ID查找雇员
public void findEmp(int id) {
//使用散列函数确定到哪条链表查找
int empLinkedListNo=hashFun(id);
Emp emp=empLinkedListArray[empLinkedListNo].findEmp(id);
if(emp!=null) {//找到
System.out.printf("在第%d条链表中找到该雇员 id=%d\n",empLinkedListNo+1,id);
}
else {
System.out.println("在哈希表中没有找到该雇员~");
}
}
public void deleteEmp(int id) {
//使用散列函数确定删除哪条链表
int empLinkedListNo=hashFun(id);
Emp emp=empLinkedListArray[empLinkedListNo].deleteEmp(empLinkedListNo+1,id);
if(emp!=null) {//找到
System.out.printf("在第%d条链表中找到该雇员 id=%d\n",empLinkedListNo+1,id);
}
else {
System.out.println("在哈希表中没有找到该雇员~");
}
}
//编写一个散列函数,使用取模法
public int hashFun(int id) {
return id%size;
}
}
class Emp{
public int id;
public String name;
public Emp next;//指向下一个节点
public Emp(int id, String name) {
super();
this.id = id;
this.name = name;
}
}
//创建EmpLinkedList,表示链表
class EmpLinkedList{
//头指针,指向第一个Emp,因此这个链表的head是直接指向第一个Emp
private Emp head;
//添加雇员到链表(加入尾部)
public void add(Emp emp) {
//如果是添加第一个雇员
if(head==null) {
head=emp;
return;
}
//如果不是添加第一个雇员,使用一个辅助指针,帮助定位到最后
Emp curEmp=head;
while(curEmp.next!=null) {
curEmp=curEmp.next;
}
curEmp.next=emp;
}
public void list(int no) {
if(head==null) {
System.out.println("第 "+no+" 条链表为空");
return;
}
System.out.print("第 "+no+" 条链表的信息为:");
Emp curEmp=head;//辅助指针
while(curEmp!=null) {
System.out.printf("=>id=%d name=%s\t",curEmp.id,curEmp.name);
curEmp=curEmp.next;
}
System.out.println();
}
//根据ID查找Emp
public Emp findEmp(int id) {
//判断链表是否为空
if(head==null) {
System.out.println("链表为空");
return null;
}
Emp curEmp=head;
while(curEmp!=null) {//假定id不重复
if(curEmp.id==id) {
break;
}
curEmp=curEmp.next;
}
return curEmp;
}
//根据id删除链表
//找到要删除链表的前一个
public Emp deleteEmp(int no,int id) {
if(head==null) {
return null;
}
Emp curEmp=head;//辅助指针
if(head.id==id) {
head=null;
System.out.printf("第%d个链表中的id=%d的节点将被删除,删除后的链表为空\n",no,id);
return curEmp;
}
while(curEmp.next!=null) {
if(curEmp.next.id==id) {
System.out.printf("第%d个链表中的id=%d的节点将被删除\n",no,id);
Emp res=curEmp.next;
curEmp=curEmp.next.next;
return res;
}
}
System.out.printf("链表中没有id=%d的雇员\n",id);
return null;
}
}