package demo;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner;
import java.util.Set;
public class Maptest {
public Map <String,Student> students;//泛型规定只能传入字符串与学生类型
public Maptest() {
this.students=new HashMap<String,Student>();//初始化
}
/*
* 测试添加学生对象,需要先检查学生是否存在,输入学生id检查学生id是否存在,不存在则添加新学生对象进students
*/
//用put方法传递键值对进Map集合
public void testPut() {
Scanner console=new Scanner(System.in);
int i=0;
System.out.println("将要添加三个学生的信息");
while(i<3){
System.out.println("请输入第"+(i+1)+"个学生的id:");
String ID=console.next();
Student st=students.get(ID);//get方法用来获取k值,如果获取的k值不存在则返回null
if(st==null) {//判断k值是否存在
System.out.println("请输入学生姓名");
String name=console.next();//获取学生姓名
Student newstudent=new Student(ID, name);//将输入的两个String值传递到新创建的学生参数里
students.put(ID, newstudent);//使用put方法将键值对传入map集合
System.out.println("成功添加学生:"+students.get(ID).name);//返回信息提示添加成功(用get获取键与键对应的值)
i++;
}
else {
System.out.println("您所输入的学生"+ID+"已存在");
continue;//结束本次循环,写不写都可以
}
}
}
/*
* 获取map集合中的所有元素。
*
* java没有自带的获取map所有键值对的方法,只能自己写
* 先用keyset方法获取map集合中的所有K键元素。
* 再用历遍get方法获取对应的V元素,再一一输出
*/
public void testkeyset() {
Set <String> keySet=students.keySet();//使用keyset方法获取map集合中的所有键值,并将键值传入set集合
System.out.println("总共有"+keySet.size()+"个学生");
for(String stuId:keySet) {
Student st=students.get(stuId);//获取map集合中的每一个k值
if(st!=null) {//当K值不为空时,打印与之相对应的学生姓名
System.out.println("学生:"+st.id+"."+st.name);
}
}
}
/*
* 利用remove方法删除map中的元素
*/
public void testRemove() {
System.out.println("请输入需要删除的学生ID!");
while (true) {
Scanner console=new Scanner(System.in);
String ID=console.next();
Student st=students.get(ID);//get方法用来获取V值,如果存入的K值不存在则返回null
if(st==null) {//判断k值是否存在
System.out.println("需要删除的学生不存在");
continue;
}
else {
students.remove(ID);
System.out.println("成功删除学生"+ID+"."+st.name);
break;
}
}
}
/*
* 获取map集合中所有的键值对
*/
public void testEntrySet() {
Set<Entry<String, Student>> entrySet=students.entrySet();
for (Entry<String, Student> entry : entrySet) {
System.out.println("取得键:"+entry.getKey()+"对应的值为:"+entry.getValue().name);
}
}
/*
* 修改map中的元素
*/
public void testModify() {
System.out.println("请输入需要修改的学生ID");
Scanner console=new Scanner(System.in);
while (true) {
String stuID=console.next();
Student student=students.get(stuID);//根据指定的k值获取V值
if(student==null) {
System.out.println("您需要修改的学生信息不存在,请重新输入");
continue;
}
else {
System.out.println("您需要修改的学生姓名为"+student.name);
System.out.println("请输入新的学生姓名");
String stuname=console.next();
Student newStudent=new Student(stuID, stuname);
students.put(stuID, newStudent);
System.out.println("修改成功");
break;
}
}
}
public static void main(String[] args) {
Maptest mt=new Maptest();
mt.testPut();//给map集合添加元素
mt.testkeyset();//展示map中的元素
mt.testRemove();//删除元素
mt.testEntrySet();//展示map中的元素
}
}