学习笔记-算法-8-Hash表

哈希表

哈希表保存雇员

public class HashTabDemo{
    public static void main(String[] args){
        // 创建HashTab
        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("exit:退出系统");
            
            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");
                    int id = scanner.nextInt();
                    hashTab.findEmpById(id);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                default:
                    break;
            }
        }
    }
}

// hash表
public HashTab{
    private EmpLinkedList[] empLinkedListArray;
    // 构造器
    public HashTab(int 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);
        // 加如对应链表
        empLinkedListArray[empLinkedListNo].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 empLinkedListNo = hashFun(id);
        Emp emp = empLinkedListArray[empLinkedListNo].findEmpById(id);
        if(emp!=null){// 找到
            System.out.printf("在%d链表中找到id=%d\n",empLinkedListNo,id);
        }else{
            System.out.println("hashtab没有找到'");
        }
    }
}

// 人员
class Emp{
    public int id;
    public String name;
    public Emp next;
    
    public Emp(int id,String name){
        this.id = id;
        this.name = name;
    }
}

// 链表
class EmpLinkedList{
    // 头
    private Emp head;
    
    // 添加人员到链表
    // 当添加雇员时，id是自增长，即id的分配是从小到大
    public void add(Emp emp){
        // 添加第一个
        if(head==null){
            head = emp;
            return;
        }
        Emp curEmp = head;
        while(true){// 移动到链表最后
            if(curEmp.next == null){
                break;
            }
            curEmp = curEmp.next;
        }
        // 退出时直接将emp加入链表
        curEmp.next = emp;
    }
    
    public void list(int no){
        if(head==null){
            System.out.println("第"+no+"链表为空");
            return;
        }
        System.out.println(第"+no+"链表为:");
        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();
    }
    
    // 查找
    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;
            }
            curEmp = curEmp.next;
        }
        return curEmp;
    }
}