백준 베스트셀러 1302번(Python, Java)

https://www.acmicpc.net/problem/1302

1302번: 베스트셀러

 

문제를 보자마자 이건 쉬울 것 같은데? 라고 생각했지만 쉽지 않았던 문제이다. ㅜㅠ 결국 풀지 못했다.

뭔가 알겠는데 파이썬과 자바가 살짝 어색해서 그런가 구현이 쉽게 되지 않았다.

 

 

1. 베스트 셀러 with Python

books = {}
for _ in range(int(input())):
    book = input()
 
    if book in books.keys():
        books[book] += 1
 
    else:
        books[book] = 1
 
 
lst=[]
for key in books.keys():
    if books[key] == max(books.values()):
        lst.append(key)
lst=sorted(lst)
print(lst[0])
 
 

 

books 라는 dictionary형으로 만들어서 key, value 를 이용해서 문제를 풀었다. 문제를 풀면서 파이썬에 대해서 새롭게 알게된 부분도 많은데 그게 바로 sorted 였다. sorted는 리스트를 정렬을 한 후에 새로운 리스트로 반환해준다.

이 문제에서 books[key] == max(books.values()) 이 부분이 뭔가 쉽게 떠오르지 않았던 부분이다. 

 

 

2. 베스트셀러 with Java

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Scanner;
 
public class BestSeller {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        HashMap<String,Integer> hm = new HashMap<>();
 
 
        int n = input.nextInt();
 
        for (int i = 0; i < n; ++i) {
            String book = input.next();
 
            if (!hm.containsKey(book)) {
                hm.put(book, 1);
            }
 
            else {
                hm.put(book, hm.get(book) + 1);
            }
        }
 
        int max = 0;
        for(String a : hm.keySet()){
            max = Math.max(max, hm.get(a));
        }
 
        ArrayList<String> list = new ArrayList<>(hm.keySet());
        Collections.sort(list);
 
        for (String a : list) {
            if (hm.get(a) == max) {
                System.out.println(a);
                break;
            }
        }
 
    }
 
}

 

자바는 확실히 파이썬보다 훨씬 구현이 까다롭다고 느꼈다..  HashMap과 ArrayList를 이용해서 풀었다. 

HashMap은 파이썬에서 dictionary 자료형과 같고, ArrayList는 리스트와 같다. 그리고 Collections.sort 는 sorted와 같아서 두 언어가 푼 개념을 똑같다. 

다만

 

ArrayList<String> list = new ArrayList<>(hm.keySet()); 을 하면 굳이 add를 하지 않아도 추가가 된다는 부분은 처음 보는데 이게 어떤 원리로 되는건지 궁금하다.

 

그래서 ArrayList가 내부적으로 어떻게 구현이 되어있는지 찾아 보니 아래와 같았다. 

  public ArrayList(Collection<? extends E> c) {
        elementData = c.toArray();
        if ((size = elementData.length) != 0) {
            // c.toArray might (incorrectly) not return Object[] (see 6260652)
            if (elementData.getClass() != Object[].class)
                elementData = Arrays.copyOf(elementData, size, Object[].class);
        } else {
            // replace with empty array.
            this.elementData = EMPTY_ELEMENTDATA;
        }
    }

 

인자로 Collection<? extends E> c) 라고 되어 있기 때문에 Collection 인터페이스에 상속받은 것들만 가능하다.

예를들면 List, Set, Map 등이 있다.