자바 알고리즘 문제 풀이 HashMap_TreeSet Q01_ 학급 회장

설명

학급 회장을 뽑는데 후보로 기호 A, B, C, D, E 후보가 등록을 했습니다.

투표용지에는 반 학생들이 자기가 선택한 후보의 기호(알파벳)가 쓰여져 있으며 선생님은 그 기호를 발표하고 있습니다.

선생님의 발표가 끝난 후 어떤 기호의 후보가 학급 회장이 되었는지 출력하는 프로그램을 작성하세요.

반드시 한 명의 학급회장이 선출되도록 투표결과가 나왔다고 가정합니다.

입력

첫 줄에는 반 학생수 N(5<=N<=50)이 주어집니다.

두 번째 줄에 N개의 투표용지에 쓰여져 있던 각 후보의 기호가 선생님이 발표한 순서대로 문자열로 입력됩니다.

출력

학급 회장으로 선택된 기호를 출력합니다.

예시 입력1	예시 출력
15 BACBACCACCBDEDE	C

 

코드

내가 입력한 코드

미완성

package HashMap_TreeSet;

import java.util.HashMap;
import java.util.Scanner;

public class Q01 {
    public int solution(int n,String str){
        int answer = 0;
        HashMap<String, Integer> map = new HashMap<>();
        
        // map.put("A",0);
        // map.put("B",0);
        // map.put("C",0);
        // map.put("D",0);
        // map.put("E",0);

        for(int i = 0;i < str.length();i++){
            if (str.indexOf(i) == 'A'){
                int result = Integer.parseInt(map.get("A").toString());
                map.put("A",++result);
            }else if(str.indexOf(i) == 'B'){
                int result = Integer.parseInt(map.get("B").toString());
                map.put("B",++result);
            }else if(str.indexOf(i) == 'C'){
                int result = Integer.parseInt(map.get("C").toString());
                map.put("C",++result);
            }else if(str.indexOf(i) == 'D'){
                int result = Integer.parseInt(map.get("D").toString());
                map.put("D",++result);
            }else if(str.indexOf(i) == 'E'){
                int result = Integer.parseInt(map.get("E").toString());
                map.put("E",++result);
            }
        }
        
        System.out.println(map.values());

        return answer;
    }

    public static void main(String[] args) {
        Q01 T = new Q01();
        Scanner kb = new Scanner(System.in);
        int n = kb.nextInt();
        String str = kb.next();
        kb.next();
        System.out.print(T.solution(n,str));
        kb.close();
    }
}

---

해설 코드

import java.util.*;
class Main {	
	public char solution(int n, String s){
		char answer=' ';
		HashMap<Character, Integer> map=new HashMap<>();
		for(char x : s.toCharArray()){
			map.put(x, map.getOrDefault(x, 0)+1); 
		}
		//System.out.println(map.containsKey('F'));
		//System.out.println(map.size());
		//System.out.println(map.remove('C'));
		
		int max=Integer.MIN_VALUE;
		for(char key : map.keySet()){
			//System.out.println(key+" "+map.get(key));
			if(map.get(key)>max){
				max=map.get(key);
				answer=key;
			}
		}
		return answer;
	}

	public static void main(String[] args){
		Main T = new Main();
		Scanner kb = new Scanner(System.in);
		int n=kb.nextInt();
		String str=kb.next();
		System.out.println(T.solution(n, str));
	}
}

 

 

풀이 및 정리 

먼저 HashMap은 key, value로 이루어져 있다.
해쉬맵 객체를 생성하고(map) map.put("key", "value")와 같이 put( ) 사용해 값을 추가할 수 있다.

해설 코드 순서를 정리해보면
입력받은 문자열 str을 x로 각각의 인덱스에 접근한다. 그리고 put( ) 메서드의 첫 번째 매개변수로 x를 넣어주고 두 번째 매개변수는 value, 즉 값이 들어가는데 str의 인덱스를 확인하면서 value 값을 증가시켜주려면 이전의 value값을 가지고 와야 한다. 

get( ) 메서드를 사용할 수 있지만 초기에 key값의 value가 존재하지 않으면 오류가 발생하므로
getOrDefault(arg1, arg2)을 사용한다. 이 메서드는 arg1에 key값이 들어가고 key값의 value를 반환한다. 하지만 해당하는 key값의 value가 존재하지 않으면 arg2를 반환한다.

HashMap 관련 메서드 정리
containsKey('key') => 매개변수 key가 HashMap에 key로 존재하는지 판단 후 boolean값 반환

size( ) => HashMap 개수를 반환

remove('key') => 매개변수 key인 key값을 삭제

keySet( ) => HashMap의 모든 key값을 가져와 반환