哈希表案例

package com.company.hashtable;

import java.util.Arrays;
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);
        EmpLinkedList empLinkedList = new EmpLinkedList();
        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("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入name");
                    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.findEmpById(id);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                default:
                    System.out.println("命令输入错误");
                    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);
    }
    //编写散列函数, 使用一个简单取模法
    public int hashFun(int id){
        return id % size;
    }
    //遍历所有的链表,遍历 hashtab
    public void list(){
        for (int i = 0; i < size; i++) {
            empLinkedListArray[i].list(i + 1);
        }
    }
    //根据 id 查找雇员
    public void findEmpById(int id){
        //使用散列函数确定到哪条链表查找
        int empLinkedListNO = hashFun(id);
        Emp emp = empLinkedListArray[empLinkedListNO].findEmpById(id);
        if (emp != null) {
            System.out.printf("在第%d条链中找到了", empLinkedListNO);
        } else {
            System.out.println("未找到");
        }
    }
}

//表示一个雇员
class Emp{
    public int id;
    public String name;
    public Emp next; //next 默认为 null
    public Emp(int id, String name){
        super();
        this.id = id;
        this.name = name;
    }
}

//创建 EmpLinkedList ,表示链表
class EmpLinkedList{
    //头指针，执行第一个 Emp,因此我们这个链表的 head 是直接指向第一个 Emp
    private Emp head;
    //添加雇员到链表
    //说明
    //1. 假定，当添加雇员时，id 是自增长，即 id 的分配总是从小到大
    // 因此我们将该雇员直接加入到本链表的最后即可
    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.println(curEmp);
            System.out.printf("=> id=%d name=%s\t",curEmp.id,curEmp.name);
            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;
            }
            if (curEmp.next == null){
                curEmp = null;
                break;
            }
            curEmp = curEmp.next;
        }
        return curEmp;
    }
}