数据结构和算法学习之哈希表实现增删查操作

package com.atguigu.HashTab;

import java.util.Scanner;

/**
 * @author 
 * @create 2022-06-14-9:01
 */
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("find:查找雇员");
            System.out.println("delete:删除雇员");
            System.out.println("exit:退出系统");
            key = scanner.next();
            switch (key) {
                case "add":
                    System.out.print("id=");
                    int id = scanner.nextInt();
                    System.out.print("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：");
                    int empId = scanner.nextInt();
                    hashTab.findEmpById(empId);
                    break;
                case "delete":
                    System.out.println("请输入要删除的雇员id：");
                    int delId = scanner.nextInt();
                    hashTab.delEmpById(delId);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                    break;
            }
        }


    }


}

//创建哈希表，管理多条链表
class HashTab {
    EmpLinkedList[] empLinkedListArray;
    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 void add(Emp emp) {
        int empLinkedNO = hashFun(emp.id);
        empLinkedListArray[empLinkedNO].add(emp);
    }

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

    }    //编写一个散列函数

    public int hashFun(int id) {
        return id % size;
    }

    //根据输入id查找雇员
    public void findEmpById(int id) {
        int empLinkedNO = hashFun(id);
        Emp emp = empLinkedListArray[empLinkedNO].findEmpById(id);
        if (emp != null) {//找到了
            System.out.printf("在第%d条链表中找到该雇员%d\n", empLinkedNO, id);

        } else {
            System.out.println("在hash表中未找到该雇员");
        }
    }

    //根据id删除雇员
    public void delEmpById(int id) {
        int empLinkedNO = hashFun(id);
        empLinkedListArray[empLinkedNO].delEmpById(id);
        System.out.println("删除成功~");
    }


}

class EmpLinkedList {
    Emp head = null;//头指针，指向当前链表的第一个雇员

    //添加员工
    public void add(Emp emp) {
        //先判断是不是第一个元素
        if (head == null) {
            head = emp;
            return;
        }
        //如果不是第一个元素，则加在最后
        Emp curEmp = head;
        while (true) {
            if (curEmp.next == null) {
                break;
            }
            curEmp = curEmp.next;
        }
        //此时curEmp指向最后一个元素
        curEmp.next = emp;
    }

    //遍历链表
    public void list(int no) {
        if (head == null) {
            System.out.println("第" + no + "条链表为空");
            return;
        }
        Emp curEmp = head;

        while (true) {
            System.out.printf("第" + no + "条链表为=>id=%d,name=%s", curEmp.id, curEmp.name);
            if (curEmp.next == null) {
                break;
            }
            curEmp = curEmp.next;
        }
        System.out.println();

    }

    //根据id查找雇员
    public Emp findEmpById(int id) {
        if (head == null) {//链表为空
            return null;
        }
        Emp curEmp = head;
        while (true) {
            if (id == curEmp.id) {//找到了
                break;
            }
            curEmp = curEmp.next;
            if (curEmp == null) {//说明找到最后也没找到
                break;
            }
        }
        return curEmp;


    }

    //根据id删除雇员
    public void delEmpById(int id) {
        boolean isFindFlag = false;
        if (head == null) {
            System.out.printf("员工号%d所在链表为空", id);
        }
        Emp temp = head;
        if (head.id == id) {//删除的如果是头节点
            head = null;
        }
        while (true) {
            if (temp.next == null) {//找到最后没找到
                break;
            }
            if (temp.next.id == id) {//找到待删除节点的前一个节点
                isFindFlag = true;
                break;
            }
            temp = temp.next;

        }
        if (isFindFlag) {
            temp.next = temp.next.next;
        }


    }
}

class Emp {
    int id;
    String name;
    Emp next;//默认为null

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