두개의 리스트에서 공통된 요소만 찾을 때, 교집합을 만드는 방법을 소개합니다.
1. retainAll()으로 두 리스트의 교집합 찾기
아래와 같이 retainAll()
으로 두 리스트의 교집합을 찾을 수 있습니다. retainAll()
는 인자로 전달된 Collection(Set or List)의 요소들만 남기고 모두 제거합니다.
결과는 Set인데, List로 변경하고 싶다면 ArrayList를 생성하면서 Set를 인자로 전달하면 됩니다.
import java.util.*;
public class Example {
public static void main(String[] args) {
List<Integer> first = Arrays.asList(1, 5, 2, 4, 6, 1);
List<Integer> second = Arrays.asList(1, 3, 5, 7);
Set<Integer> common = new HashSet<>(first);
common.retainAll(second);
System.out.println(common);
// Set to List
List<Integer> commonList = new ArrayList<>(common);
System.out.println(commonList);
}
}
Output:
[1, 5]
[1, 5]
2. Stream과 filter()로 두 리스트의 교집합 찾기
아래와 같이 Stream과 filter()
로 두 리스트의 교집합을 찾을 수 있습니다. filter()에서 contains()로 second 리스트가 갖고 있는 요소만 Set에 추가합니다.
import java.util.*;
import java.util.stream.Collectors;
public class Example1 {
public static void main(String[] args) {
List<Integer> first = Arrays.asList(1, 5, 2, 4, 6, 1);
List<Integer> second = Arrays.asList(1, 3, 5, 7);
Set<Integer> common = first.stream().filter(second::contains).collect(Collectors.toSet());
System.out.println(common);
// Set to List
List<Integer> commonList = new ArrayList<>(common);
System.out.println(commonList);
}
}
Output:
[1, 5]
[1, 5]
Loading script...
Related Posts
- Java - compareTo(), 객체 크기 비교
- Java - BufferedReader로 파일 읽기
- Java - BufferedWriter로 파일 쓰기
- Java - BigInteger 범위, 비교, 연산, 형변환
- Java charAt() 함수 알아보기
- 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 배열 초기화 방법