Map에서 Key, Value 다루기

 알고리즘 문제를 풀다보면 map으로 해결해야 하는 상황이 많습니다. 또한 map을 적절히 이용해 문제가 원하는 key값과 value값을 꺼낼 수 있어야 합니다. 빠르고 쉽게 값을 뽑아내는 방법을 알아보겠습니다. 다음의 조건이 주어질 때 어떤 방식을 사용하는 것이 효율적일까요? 개인적으로 iterator를 순회하는 것보다 for문으로 조회하는 것을 선호합니다.

 

Map<Integer,Integer> 기준

 

 

*변형 내용

1. Map의 요소들을 순회하는 방법은? ( 1. key 순회 2.value 순회 3. key , value 동시 순회)

2. Map의 key 값, value 값 중 최대 값을 뽑는 방법은?

3. Map의 value값 중 최대 값의 key값을 뽑는 방법은? ( + 중복일 경우도 포함)

 

 

1번

import java.util.HashMap;
import java.util.Map;

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Map<Integer,Integer> map = new HashMap<>();
		map.put(4, 2);
		map.put(1, 3);
		map.put(5, 3);
		map.put(2, 3);
		
	
		for(Integer key : map.keySet()) {
			System.out.println(key);
		}
		
		for(Integer value : map.values()) {
			System.out.println(value);
		}
		
		for(Map.Entry<Integer, Integer> m : map.entrySet()) {
			System.out.println(m.getKey()+","+m.getValue());
		}
	}
}

 

key를 순회할 때는 keySet()

value를 순회할 때는 values()

key, value를 순회할 때는 entrySet()

 

참고로 다른 형태는 보다시피 Set()형태로 반환하는데 values()의 반환형은 Collection입니다.

Collection은 List, Set, Queue와 같은 Collection interface의  클래스들 중 하나 입니다. add(), remove(), clear(), size() 등으 메소드 사용이 가능합니다.

 

Collections는 Collection을 조작할 때 사용하는 유용한 클래스로 sort()가 대표적이며 min(), max()도 있습니다.

 

 

2번

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Map<Integer,Integer> map = new HashMap<>();
		map.put(4, 2);
		map.put(1, 3);
		map.put(5, 3);
		map.put(2, 3);
		System.out.println(Collections.max(map.values()));
		System.out.println(Collections.min(map.values()));
	}
}

 

Collections.max() , Collections.min()을 사용

 

 

3번

 

최대 value값이 1개라면 간단한 방법으로 key값이 조회가 가능하지만,

Collections.max(map.entrySet(), (m1, m2) -> m2.getValue() - m1.getValue()).getKey()

 

 

value값이 중복이고, 그 중에서 가장 큰 key값을 찾으려면 stream을 사용해야 합니다. (간단한 방법)

 

map.entrySet().stream().max((m1, m2) -> m1.getValue() > m2.getValue() ? 1 : -1).get().getKey()

 

특히 최대 value값이 중복이고 모든 key값을 조회하고 싶다면, maxValue를 찾고 다시 Map을 순회해아 합니다.

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Map<Integer, Integer> map = new HashMap<>();
		map.put(4, 2);
		map.put(1, 3);
		map.put(5, 3);
		map.put(2, 3);

		int maxValue = Collections.max(map.values());
		for(Map.Entry<Integer, Integer> m : map.entrySet()) {
			if(m.getValue()==maxValue) {
				System.out.println(m.getKey());
			}
		}
	}
}