12.学生选课---删除Map中的映射

package com.imooc.collection;

import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner;
import java.util.Set;

public class MapTest {
    /**
     * 用来承装学生类型对象
     *
     * @param args
     */
    public Map<String, Student> students;

    /**
     * 在构造器中初始化students属性
     *
     * @param args
     */
    public MapTest() {
        this.students = new HashMap<String, Student>();
    }

    /**
     * 测试添加：输入学生ID，判断是否有被占用，若未被占用，则输入姓名，创建新学生对象，并且添加到students中
     *
     * @param args
     */

    public void testPut() {
        // 创建一个 Scanner对象，用来获取输入的学生ID和姓名
        Scanner console = new Scanner(System.in);
        int i = 0;
        while (i < 3) {
            System.out.println("请输入学生ID：");
            String ID = console.next();
            // 判断ID是否被占用
            Student st = students.get(ID);
            if (st == null) {
                // 提示输入学生姓名
                System.out.println("请输入学生姓名：");
                String name = console.next();
                // 创建新的学生对象
                Student newStudent = new Student(ID, name);
                // 通过调用students的put方法，添加ID-学生映射
                students.put(ID, newStudent);
                System.out.println("成功添加学生 ：" + students.get(ID).name);
                i++;
            } else {
                System.out.println("该学生ID已被占用。");
                continue;
            }
        }
    }

    /**
     * 测试Map的keySet方法
     *
     * @param args
     */

    public void testKeySet() {
        // 通过keySet方法，返回Map中的所有“键”的Set集合
        Set<String> keySet = students.keySet();
        // 取得students的容量
        System.out.println("总共有" + students.size() + "个学生");
        // 遍历keySet，取得每一个键，在调用get方法取得每个键对应的value
        for (String stuId : keySet) {
            Student st = students.get(stuId);
            if (st != null)
                System.out.println("学生：" + st.name);
        }
    }

    /**
     * 测试删除Map中的映射
     *
     * @param args
     */

    public void testRemove() {

        // 获取从键盘输入的待删除学生ID字符串
        Scanner console = new Scanner(System.in);
        while (true) {
            System.out.println("请输入要删除的学生ID！");
            String ID = console.next();
            // 判断该ID是否有对应的学生对象
            Student st = students.get(ID);
            if (st == null) {
                // 提示输入的ID并不存在
                System.out.println("该ID不存在。");
                continue;
            }
            students.remove(ID);
            System.out.println("成功删除学生：" + st.name);
            break;
        }

    }

    /**
     * 通过entrySet方法来遍历Map
     *
     * @param args
     */

    public void testEntrySet() {
        // 通过entrySet方法返回Map中的所有键值对
        Set<Entry<String, Student>> entrySet = students.entrySet();
        for (Entry<String, Student> entry : entrySet) {
            System.out.println("取得键：" + entry.getKey());
            System.out.println("取得对应的值为：" + entry.getValue().name);
        }
    }

    public static void main(String[] args) {
        MapTest mt = new MapTest();
        mt.testPut();
        mt.testKeySet();
        mt.testRemove();
        mt.testEntrySet();
    }

}