(22.10.24)Java프로그래밍:HashSet 정렬 방법과 Map(HashMap)

HashSet의 중복을 피해서 저장하기

학습과정(Course)을 관리하는 기능 작성

Course 과정명(name), 개설연도(year), 차수(num), 담당교사(teacher)가 있고 이중 과정명, 개설연도, 차수가 동일하면 동일과정으로  간주한다. 키보드에서 속성값을 입력받아서 Course객체를 초기화 한다.

동일한 객체가 중복되지 않도록 한다. 입력된 Course 내용을 화면에 목록으로 표시한다.

 

Main 클래스 코드

import java.util.HashSet;
import java.util.Scanner;
import java.util.Iterator;
import com.ezen.javaoop.Course;
public class ExceptionMain 
{
    public static void main(String[] args) 
    {
        HashSet<Course> colist = new HashSet<>();
        for(int i=0;i<5;i++) {
            Scanner kbd = new Scanner(System.in);
            System.out.print("과정명을 입력하세요.");
            String coname = kbd.nextLine().strip();
            System.out.print("개설연도를 입력하세요.");
            int coyear = kbd.nextInt();
            kbd.nextLine();
            System.out.print("차수를 입력하세요.");
            int conum = kbd.nextInt();
            kbd.nextLine();
            System.out.print("담당교사를 입력하세요.");
            String coteacher = kbd.nextLine().strip();
        }
        Iterator<Course> iter =  colist.iterator();
        while(iter.hasNext()) {
            Course a = iter.next();
            System.out.println(a);
        }
    }
}

 

Course 클래스 코드

import java.util.Objects;
 
public class Course 
{
    private String name;
    private int year;
    private int num;
    private String teacher;
    public Course() {}
    public Course(String name, int year, int num, String teacher) {
        this.setName(name);
        this.setYear(year);
        this.setNum(num);
        this.setTeacher(teacher);
    }
    @Override
    public boolean equals(Object obj) {
        Course other = (Course)obj;
        return  this.name.equals(other.name)&&
                this.year==(other.year)&&
                this.num==other.num;
    }
    @Override
    public String toString() {
        String print = String.format("%s\t%d\t%d\t%s", name, year, num, teacher);
                
        return print;
    }
    @Override
    public int hashCode() {
        
        return Objects.hash(name,year,num);
    }
}

콘솔 값:

과정명을 입력하세요.자바
개설연도를 입력하세요.2000
차수를 입력하세요.1
담당교사를 입력하세요.김뭐시기
과정명을 입력하세요.자바
개설연도를 입력하세요.2000
차수를 입력하세요.1
담당교사를 입력하세요.홍길동
과정명을 입력하세요.파이썬
개설연도를 입력하세요.2010
차수를 입력하세요.2
담당교사를 입력하세요.김옥분
과정명을 입력하세요.파이썬
개설연도를 입력하세요.2012
차수를 입력하세요.1
담당교사를 입력하세요.진달래
과정명을 입력하세요.자바
개설연도를 입력하세요.2000
차수를 입력하세요.1
담당교사를 입력하세요.황순원
파이썬 2010 2 김옥분
파이썬 2012 1 진달래
자바 2000 1 김뭐시기

 

"자바 2000 1 김뭐시기"와 과정명, 개설연도, 차수이 같은 "자바 2000 1 황순원"과 "자바 2000 1 홍길동"은 빼고 3가지 항목만 출력된 걸 확인할 수 있다. Hash에서 내용이 같음을 비교하기 위해서 equals() 메소드를 오버라이드 해줄땐 항상 hashcode도 같이 오버라이드 해줘야한다.

---

HashSet을 순서대로 나타내주고 싶으면 ArrayList로 먼저 바꾸어 줘야한다. 왜냐하면 HashSet은 순서를 지켜주지 않기 때문이다. ArrayList로 바꿔주고 Collection.sort() 기능을 쓰면 순서대로 나열할 수 있다. 하지만 이때 sort에 에러 표시가 뜨는데 이는 자바에서 지원해주는 String 클래스 같은 것이 아닌 작성자가 만든 Course 클래스 이기 때문에 Course클래스에서 어떠한 순서로 나열할 것인지 정해줘야한다.

 

Main 클래스 코드

import java.util.HashSet;
import java.util.Iterator;
import com.ezen.javaoop.Course;
import java.util.ArrayList;
import java.util.Collections;
 
public class ExceptionMain 
{
    public static void main(String[] args) 
    {
        HashSet<Course> colist = new HashSet<>();
        
        colist.add(new Course("자바",2020,1,"홍길동"));
        colist.add(new Course("파이썬",2020,1,"홍길동"));
        colist.add(new Course("자바",2020,1,"김장미"));
        colist.add(new Course("자바",2022,1,"진달래"));
        colist.add(new Course("자바",2022,2,"고길동"));
        colist.add(new Course("파이썬",2020,2,"황순원"));
        
        Iterator<Course> iter =  colist.iterator();
        while(iter.hasNext()) {
            Course a = iter.next();
            System.out.println(a);    
        }
        System.out.println("정렬후");
        ArrayList<Course> arcolist = new ArrayList<>(colist);
        Collections.sort(arcolist);
        for(int i=0;i<arcolist.size();i++) {
            System.out.println(arcolist.get(i));
        }
    }
}

Course 클래스 코드

import java.util.Objects;
 
public class Course implements Comparable<Course>
{
    private String name;
    private int year;
    private int num;
    private String teacher;
    public Course() {}
    public Course(String name, int year, int num, String teacher) {
        this.setName(name);
        this.setYear(year);
        this.setNum(num);
        this.setTeacher(teacher);
    }
    
    @Override
    public boolean equals(Object obj) {
        Course other = (Course)obj;
        return  this.name.equals(other.name)&&
                this.year==(other.year)&&
                this.num==other.num;
    }
    @Override
    public String toString() {
        String print = String.format("%s\t%d\t%d\t%s", name, year, num, teacher);
                
        return print;
    }
    @Override
    public int hashCode() {
        
        return Objects.hash(name,year,num);
    }
    @Override
    public int compareTo(Course o) {
        // 현재 객체가 더 작다고 표현하려면 -1리턴(정렬 상 먼저 위치)
        //현재 객체가 더 크다고 표현하려면 1을 리턴(정렬 상 나중 위치)
        //현재 객체와 동일한 크기라고 표현하려면 0을 리턴
        if(this.year<o.year) return -1;
        else if(this.year>o.year)return 1;
        else {
            if(this.num<o.num) return-1;
            else if(this.num<o.num) return 1;
            else return 0;
        }
    }
}

콘솔 값:

파이썬 2020 1 홍길동
파이썬 2020 2 황순원
자바 2022 1 진달래
자바 2022 2 고길동
자바 2020 1 홍길동
정렬후
파이썬 2020 1 홍길동
자바 2020 1 홍길동
파이썬 2020 2 황순원
자바 2022 1 진달래
자바 2022 2 고길동

 

6개의 정보를 추가해 줬지만 중복된 1가지는 나오지 않았다. 정렬 해주지 않고 그냥 HashSet을 출력했을 땐 "자바 2020 1 홍길동"이 정렬되지 않은채 5번째 나오걸 볼 수 있지만, 연도별로 먼저 정렬해주고 연도가 같다면 차수별로 정렬해주는 compare To() 메소드를 오버라이드 해주니깐 정렬된 값을 얻을 수 있었다.

---

●Map●

key 와 value를 쌍으로 저장한다.

주로쓰는 자료구조로는 HashMap<○○○,●●●> 과 HashTable이 있다.

순차 검색을 하지 않아도 돼서, 가장 빠른 검색기능을 가지고 있다. 처음부터 끝까지 쭉 확일할 필요가 없기 때문이다.

import java.util.HashMap;
public class ExceptionMain 
{
    public static void main(String[] args) 
    {
        HashMap<String,String> map = new HashMap<>();
        map.put("a", "해린");
        map.put("b", "하니");
        map.put("c", "민지");
        map.put("c", "혜인");
        map.put("e", "다니엘");
        
        String name = map.get("a");
        System.out.println(name);
        String name2 = map.get("c");
        System.out.println(name2);
    }
}

콘솔 값:

해린

혜인

 

키는 중복되지 않고 value가 덮어써졌기 때문에 c-민지 대신에 c-혜인 값이 나중에 저장되므로 혜인이 출력된다.

---

반복문을 쓰기 위해선 keyset()을 이용하여 Map을 Set 형태로 바꾸고 반복자를 이용하여 반복문을 쓰면된다.

import java.util.HashMap;
public class ExceptionMain 
{
    public static void main(String[] args) 
    {
        HashMap<String,String> map = new HashMap<>();
        map.put("a", "해린");
        map.put("b", "하니");
        map.put("c", "민지");
        map.put("c", "혜인");
        map.put("e", "다니엘");
        
        Set<String> keyset = map.keySet();
        Iterator<String> iter = keyset.iterator();
        while(iter.hasNext()) {
            String key = iter.next();
            String val = map.get(key);
            System.out.println(val);
        }
    }
}

콘솔 값:

해린
하니
혜인
다니엘

---

전부 삭제 => .clear

수정하기=> .replace()

키에서 검색=> .containsKey()

값에서 검색 =>.containValue()

삭제=> .remove()

키빼고 값만 목록으로 표시=> .values()

---

Map을 활용한 CRUD 실습

User (uid, name, email, phone)

Map에 key, value를 저장할 때 key로서 uid를 사용하고 value에는 User 객체의 참조를 저장한다.

HashMap을 이용하여 CRUD를 만들어 보아라

목록을 표시할때는 name 순으로 정렬되어야 한다.

 

Main 클래스 코드

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Scanner;
import com.ezen.javaoop.User;
 
public class Main2 
{
    static enum Menu { LIST, ADD, FIND, UPDATE, DELETE, EXIT};
    public static void main(String[] args) 
    {
         HashMap<String,User> map = new HashMap<>();
         Menu menu;
            boolean go = true;
            while(go){
                menu = menu();
                switch(menu)
                {
                case LIST:  list(map);        break;
                case ADD:   add(map);        break;
                case FIND:  find(map);        break;
                case UPDATE:update(map);    break;
                case DELETE:delete(map);    break;
                case EXIT:  go=false;        break;
                default: System.out.println("메뉴입력 오류");
                }
            }
            System.out.println("프로그램 종료");
    }
    private static Menu menu(){
        Scanner kbd = new Scanner(System.in);
        System.out.print("목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):");
        String m = kbd.nextLine().strip();
        Menu menu = null;
        switch(m)
        {
            case "s": menu = Menu.LIST;     break;
            case "a": menu = Menu.ADD;      break;
            case "f": menu = Menu.FIND;     break;
            case "u": menu = Menu.UPDATE;   break;
            case "d": menu = Menu.DELETE;   break;
            case "x": menu = Menu.EXIT;     break;
            default : menu = null; 
        }
        return menu;
    }
    private static void add(HashMap<String,User> map){
        Scanner kbd = new Scanner(System.in);
        
        System.out.print("아이디:");
        String uid = kbd.nextLine().strip();
        System.out.print("이름:");
        String name = kbd.nextLine().strip();
        System.out.print("이메일:");
        String email = kbd.nextLine().strip();
        System.out.print("전화번호:");
        int phone = kbd.nextInt();
        kbd.nextLine();
 
        map.put(uid, new User(uid, name,email,phone));
    }
    private static String getKeyword(){
        Scanner kbd = new Scanner(System.in);
        System.out.print("검색할 ID는 무엇입니까?");
        String newid = kbd.nextLine().strip();
        return newid;
    }
    private static void list(HashMap<String,User> map){
        Collection<User> coll = map.values();
        ArrayList<User> list = new ArrayList<>(coll);
        Collections.sort(list);
        System.out.println();
        System.out.println("아이디\t이름\t이메일\t전화번호");
        for(int i=0;i<list.size();i++) {
            System.out.println(list.get(i));
        }
    }
    private static void find(HashMap<String,User> map) {
        String key = getKeyword();
        if(map.containsKey(key)){
            System.out.println("아이디\t이름\t이메일\t전화번호");
            System.out.println(map.get(key));
        }else {
            System.err.println("찾는 사용자 정보가 없습니다.");
        }
    }
    private static void update(HashMap<String,User> map) {
        String key = getKeyword();
        Scanner kbd = new Scanner(System.in);
        if(map.containsKey(key)){
            System.out.println("검색성공");
            System.out.println("수정할 이름:");
            String name = kbd.nextLine().strip();
            System.out.println("수정할 이메일:");
            String email = kbd.nextLine().strip();
            System.out.println("수정할 전화번호:");
            int phone = kbd.nextInt();
            kbd.nextLine();
            map.replace(key, new User(key, name,email,phone));
        }
    }
    private static void delete(HashMap<String,User> map){
        String key = getKeyword();
        if(map.containsKey(key)){
            map.remove(key);
            System.out.println("삭제 성공");
        }else {
            System.out.println("삭제 대상 검색 실패");
        }
    }
}
 

User 클래스 코드

public class User implements Comparable<User>
{
    private String uid;
    private String name;
    private String email;
    private int phone;
    
    public User() {}
    public User(String uid, String name, String email, int phone) {
        this.uid = uid;
        this.name = name;
        this.email = email;
        this.phone = phone;
    }
    @Override
    public String toString() {
        String print = String.format("%s\t%s\t%s\t%d",uid, name, email,phone);    
        return print;
    }
    @Override
    public int compareTo(User other) {
        return this.uid.compareTo(other.uid);    
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
    public String getEmail() {
        return email;
    }
    public void setEmail(String email) {
        this.email = email;
    }
    public int getPhone() {
        return phone;
    }
    public void setPhone(int phone) {
        this.phone = phone;
    }
}

콘솔 값: 

목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):a
아이디:kkk
이름:paul
이메일:bjdf@nad.com
전화번호:546512
목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):a
아이디:aaa
이름:bob
이메일:fsko@nad.com
전화번호:415475
목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):s

아이디 이름 이메일 전화번호
aaa bob fsko@nad.com 415475
kkk paul bjdf@nad.com 546512
목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):f
검색할 ID는 무엇입니까?aaa
아이디 이름 이메일 전화번호
aaa bob fsko@nad.com 415475
목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):u
검색할 ID는 무엇입니까?aaa
검색성공
수정할 이름:
jun
수정할 이메일:
fwkfl@nad.com
수정할 전화번호:
6845155
목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):s

아이디 이름 이메일 전화번호
aaa jun fwkfl@nad.com 6845155
kkk paul bjdf@nad.com 546512
목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):d
검색할 ID는 무엇입니까?kkk
삭제 성공
목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):s

아이디 이름 이메일 전화번호
aaa jun fwkfl@nad.com 6845155
목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):x
프로그램 종료