[Java] Softeer 금고 털이 (HashMap)

[Java] Softeer 금고 털이 (HashMap)

문제 링크 : https://softeer.ai/practice/6288

Softeer - 현대자동차그룹 SW인재확보플랫폼

 

 

 

기록의 이유...

사실 알고리즘을 생각하는 데에는 그렇게까지 오랜 시간이 걸리진 않았다

처음엔 0-1 냅색 기준으로 좀 떠올리려고 했는데 물 떠오면서 생각해보니 그냥 그럴 필요가 없었던ㅋ

대신 파이썬으로 하면 이렇게이렇게 하고 되게 금방 풀었을텐데...

자바로 하려니 하나씩 검색하고 작성하는 게 힘들었다 ㅠㅠ 어렵다 어려워~~

 

HashMap

파이썬의 dictionary 같은 역할이라고 생각하면 된다.

마찬가지로 key, value값이 존재한다. 

 

자바에 map이랑 hashmap으로 구분이 되어있는 것 같은데 (차이는 나도 잘 모름)

map이 부모라 Map<자료형, 자료형> 변수명 = HashMap 어쩌구도 되는 것 같다

 

map에 값을 넣을 땐 put, 값을 꺼내올 땐 get으로 사용하면 된다.

map.put(key, value)

map.get(key) ==> return value 형태

 

다만, put을 할 때 이미 key에 대한 값이 존재한다면 당연히 엎어치기가 된다.

이 점에 유의해서 코드를 짜야할듯.

그래서 이 문제에서 만약 해당 키 값이 해시맵에 존재한다면 을 표현하기 위해

map.containsKey(key) 로 확인했다. 존재한다면 true, 존재하지 않는다면 false로.

 

hashmap 정렬도 된다.

물론 출력값으로 보는 방법은 아직 모르겠지만^^

key값만 정렬하는 방법은 hashmap의 key값들만 list 형태로 가지고 나와서 정렬하는 방법이 있다.

python에서는 dict.keys(), dict.values()로 했다면, 자바에서는 dict.keySet(), dict.valueSet()으로 한다.

 

List<type> keys = new ArrayList<>(map.keySet()); 으로 listtype으로 꺼내오고

Collections.sort(keys)로 내장함수 사용하여 keys 배열 정렬.

만약 내림차순으로 하고 싶다면, Collections.reverse(keys) 사용해서 그대로 뒤집어주기.

사실 내림차순 정렬이라기보다는, 정렬 > list 뒤집어줌 으로써 내림차순 효과처럼 보이게 하는 것 같다.

 

 

이 한 문제에서 이렇게 검색하고 알아볼 것들이 많았다니,,,

좀 부끄럽지만 그래도 찾아보고 문제 풀고 맞히니 오랜만에 재미있었다 ㅎㅎ

 

제출 코드 (정답)

import java.io.*;
import java.util.*;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int weight = sc.nextInt();
        int types = sc.nextInt();

        HashMap<Integer, Integer> valdict = new HashMap<Integer, Integer>(types);
        for(int i=0;i<types;i++){
            int m = sc.nextInt();
            int p = sc.nextInt();

            if (valdict.containsKey(p)){
                int tmp = valdict.get(p);
                valdict.put(p, tmp+m);
            }
            else {
                valdict.put(p, m);
            }
        }

        List<Integer> keys = new ArrayList<>(valdict.keySet());
        Collections.sort(keys);
        Collections.reverse(keys);
        
        int result = 0;
        for (Integer key:keys){
            if(valdict.get(key) <= weight){
                result += key*valdict.get(key);
                weight -= valdict.get(key);
            }
            else {
                result += key*weight;
                weight = 0;
            }
        }

        System.out.println(result);
    }
}