for, while 등, 반복문 안에서 HashMap을 순회하면서 요소를 삭제하는 방법을 소개합니다.
1. ConcurrentModificationException
반복문을 이용하여 HashMap을 순회하면서 특정 조건의 요소를 삭제하는 경우가 있습니다.
문제는 HashMap 순회 중, 요소를 삭제하면 ConcurrentModificationException가 발생할 수 있습니다.
예를 들어, 아래 처럼 HashMap을 순회하면서 HashMap.remove()
로 요소를 삭제하면 Exception이 발생할 수 있습니다.
따라서, 아래와 같은 방식으로 삭제하면 안됩니다.
import java.util.HashMap;
public class Example {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
map.put("apple", 100);
map.put("kiwi", 200);
map.put("banana", 300);
map.put("grape", 400);
for (String key: map.keySet()) {
if (key.startsWith("k")) {
map.remove(key);
}
}
System.out.println(map);
}
}
Output:
Exception in thread "main" java.util.ConcurrentModificationException
at java.base/java.util.HashMap$HashIterator.nextNode(HashMap.java:1597)
at java.base/java.util.HashMap$KeyIterator.next(HashMap.java:1620)
at Example.main(Example.java:13)
2. Iterator로 순회 중 요소 삭제
Iterator를 이용하여 순회하면서 Iterator.remove()
로 요소를 삭제하는 것은 Exception이 발생하지 않습니다.
아래와 같이 특정 요소를 삭제하도록 구현할 수 있습니다.
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
public class Example {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
map.put("apple", 100);
map.put("kiwi", 200);
map.put("banana", 300);
map.put("grape", 400);
Iterator<Map.Entry<String, Integer>> it = map.entrySet().iterator();
while (it.hasNext()) {
if (it.next().getKey().startsWith("k")){
it.remove();
}
}
System.out.println(map);
}
}
Output:
{banana=300, apple=100, grape=400}
3. List를 이용하여 HashMap 요소 삭제
HashMap 순회 중 삭제할 요소를 리스트에 저장하고, List를 다시 순회하면서 HashMap의 요소를 삭제할 수 있습니다. HashMap 순회 중 요소를 삭제하는 것이 아니기 때문에 ConcurrentModificationException이 발생하지 않습니다.
아래와 같이 요소 삭제를 할 수 있습니다.
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
public class Example {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
map.put("apple", 100);
map.put("kiwi", 200);
map.put("banana", 300);
map.put("grape", 400);
List<String> willRemove = new ArrayList<>();
for (String key: map.keySet()) {
if (key.startsWith("k")) {
willRemove.add(key);
}
}
for (String str : willRemove) {
map.remove(str);
}
System.out.println(map);
}
}
Output:
{banana=300, apple=100, grape=400}
4. removeIf()로 순회 중 요소 삭제
removeIf()
는 인자로 인자 1개와 리턴 값이 있는 함수형 인터페이스를 받습니다. 이 함수에서 어떤 요소에 대해서 true를 리턴하면, 그 요소는 삭제됩니다.
removeIf()
도 ConcurrentModificationException을 발생시키지 않기 때문에, 이 방법으로 요소를 삭제할 수 있습니다.
아래와 같이 특정 요소를 삭제하도록 구현할 수 있습니다.
import java.util.HashMap;
public class Example {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
map.put("apple", 100);
map.put("kiwi", 200);
map.put("banana", 300);
map.put("grape", 400);
map.entrySet().removeIf(entry -> entry.getKey().startsWith("k"));
System.out.println(map);
}
}
Output:
{banana=300, apple=100, grape=400}
4.1 HashMap.keySet().removeIf()
entrySet()
대신에 keySet()
을 사용하여 요소를 삭제할 수도 있습니다.
keySet()
은 HashMap의 모든 key에 대한 collection이기 때문에, key만 순회할 수 있습니다.
import java.util.HashMap;
public class Example {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
map.put("apple", 100);
map.put("kiwi", 200);
map.put("banana", 300);
map.put("grape", 400);
map.keySet().removeIf(key -> key.startsWith("k"));
System.out.println(map);
}
}
Output:
{banana=300, apple=100, grape=400}
4.2 HashMap.values().removeIf()
values()
는 HashMap의 value들을 collection으로 리턴하며, removeIf()
로 특정 value에 대한 요소를 삭제할 수 있습니다.
import java.util.HashMap;
public class Example {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
map.put("apple", 100);
map.put("kiwi", 200);
map.put("banana", 300);
map.put("grape", 400);
map.values().removeIf(val -> val == 200);
System.out.println(map);
}
}
Output:
{banana=300, apple=100, grape=400}
Related Posts
- Java - Unsupported class file major version 61 에러
- Java - String.matches()로 문자열 패턴 확인 및 다양한 예제 소개
- Java - 문자열 공백제거 (trim, replace)
- Java - replace()와 replaceAll()의 차이점
- Java - ArrayList 초기화, 4가지 방법
- Java - 배열 정렬(Sorting) (오름차순, 내림차순)
- Java - 문자열(String)을 비교하는 방법 (==, equals, compare)
- Java - StringBuilder 사용 방법, 예제
- Java - 로그 출력, 파일 저장 방법 (Logger 라이브러리)
- Java IllegalArgumentException 의미, 발생 이유
- Java - NullPointerException 원인, 해결 방법
- Seleninum의 ConnectionFailedException: Unable to establish websocket connection 해결
- Java - compareTo(), 객체 크기 비교
- Java - BufferedWriter로 파일 쓰기
- Java - BufferedReader로 파일 읽기
- Java charAt() 함수 알아보기
- Java - BigInteger 범위, 비교, 연산, 형변환
- Java contains()로 문자(대소문자 X) 포함 확인
- Java - Set(HashSet)를 배열로 변환
- Java - 문자열 첫번째 문자, 마지막 문자 확인
- Java - 문자열 한글자씩 자르기
- Java - 문자열 단어 개수 가져오기
- Java - 1초마다 반복 실행
- Java - 배열을 Set(HashSet)로 변환
- Java - 여러 Set(HashSet) 합치기
- Java - 명령행 인자 입력 받기
- Java - 리스트 역순으로 순회, 3가지 방법
- Java - 특정 조건으로 리스트 필터링, 3가지 방법
- Java - HashMap 모든 요소들의 합계, 평균 계산
- Java - 특정 조건으로 HashMap 필터링
- Java - 싱글톤(Singleton) 패턴 구현
- Java - 숫자 왼쪽에 0으로 채우기
- Java - String 배열 초기화 방법
- Java - 정렬된 순서로 Map(HashMap) 순회
- Java - HashMap에서 key, value 가져오기