3-2 공통원소 구하기 (Java)

정우·2022년 10월 8일

인프런코테

인프런 기초코테 공부

목록 보기

24/25

✏️ 문제

설명

A, B 두 개의 집합이 주어지면 두 집합의 공통 원소를 추출하여 오름차순으로 출력하는 프로그램을 작성하세요.

입력

첫 번째 줄에 집합 A의 크기 N(1<=N<=30,000)이 주어집니다.

두 번째 줄에 N개의 원소가 주어집니다. 원소가 중복되어 주어지지 않습니다.

세 번째 줄에 집합 B의 크기 M(1<=M<=30,000)이 주어집니다.

네 번째 줄에 M개의 원소가 주어집니다. 원소가 중복되어 주어지지 않습니다.

각 집합의 원소는 1,000,000,000이하의 자연수입니다.

출력

두 집합의 공통원소를 오름차순 정렬하여 출력합니다.

예제입력
5
1 3 9 5 2
5
3 2 5 7 8

예제출력
2 3 5

✏️ 코드


import java.util.Scanner;
import java.util.ArrayList;
import java.util.Arrays;

class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] arr = new int[n];
        for (int i=0; i<n; i++) {
            arr[i] = sc.nextInt();
        }
        int m = sc.nextInt();
        int[] arr1 = new int[m];
        for (int i=0; i<m; i++) {
            arr1[i] = sc.nextInt();
        }
        sc.close();

        for (int i : solution(n, m, arr, arr1)) {
            System.out.print(i + " ");
        }
    }

    public static ArrayList<Integer> solution(int n, int m, int[] arr, int[] arr1) {
        ArrayList<Integer> answer = new ArrayList<Integer>();
        Arrays.sort(arr);
        Arrays.sort(arr1);
        int p1 = 0, p2 = 0;

        while (p1<n && p2<m) {
            if (arr[p1] == arr1[p2]) {
                answer.add(arr[p1]);
                p1++;
                p2++;
            }
            else if (arr[p1] < arr1[p2]) {
                p1++;
            }
            else {
                p2++;
            }
        }

        return answer;
    }
}