[JAVA71] 1269. 대칭 차집합

1269. 대칭 차집합 (S4)

시간 제한	메모리 제한	제출	정답	맞힌 사람	정답 비율
2 초	256 MB	36912	23747	19693	64.927 %

 

문제

자연수를 원소로 갖는 공집합이 아닌 두 집합 A와 B가 있다. 이때, 두 집합의 대칭 차집합의 원소의 개수를 출력하는 프로그램을 작성하시오. 두 집합 A와 B가 있을 때, (A-B)와 (B-A)의 합집합을 A와 B의 대칭 차집합이라고 한다.

예를 들어, A = { 1, 2, 4 } 이고, B = { 2, 3, 4, 5, 6 } 라고 할 때,  A-B = { 1 } 이고, B-A = { 3, 5, 6 } 이므로, 대칭 차집합의 원소의 개수는 1 + 3 = 4개이다.

입력

첫째 줄에 집합 A의 원소의 개수와 집합 B의 원소의 개수가 빈 칸을 사이에 두고 주어진다. 둘째 줄에는 집합 A의 모든 원소가, 셋째 줄에는 집합 B의 모든 원소가 빈 칸을 사이에 두고 각각 주어진다. 각 집합의 원소의 개수는 200,000을 넘지 않으며, 모든 원소의 값은 100,000,000을 넘지 않는다.

출력

 

첫째 줄에 대칭 차집합의 원소의 개수를 출력한다.

---

summary

대칭 차집합의 원소의 개수 구하기

 

strategy

투포인터 

• 두 집합을 정렬함 (Collections.sort)
• 각 원소를 비교함 (AList.get(Aidx))
• 두 원소 중 더 작은값의 인덱스를 이동 (Aidx, Bidx)
• 두 원소가 같은 경우 제외 시킴 (Aanswer, Banswer)

 

note

• 원소의 개수는 200,000 넘지 않음
• 모든 원소의 값은 100,000,000을 넘지 않음 (long)

 

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

public class Main {
    static int A, B;
    static List<Integer> AList = new ArrayList<>(), BList = new ArrayList();
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        StringTokenizer input = new StringTokenizer(br.readLine());
        A = Integer.parseInt(input.nextToken());
        B = Integer.parseInt(input.nextToken());

        input = new StringTokenizer(br.readLine());
        for(int a = 0; a < A; a++) {
            AList.add(Integer.parseInt(input.nextToken()));
        }

        input = new StringTokenizer(br.readLine());
        for(int b = 0; b < B; b++) {
            BList.add(Integer.parseInt(input.nextToken()));
        }

        Collections.sort(AList);
        Collections.sort(BList);

        int Aanswer = AList.size(), Banswer = BList.size();
        int Aidx = 0, Bidx = 0;
        while(Aidx < AList.size() && Bidx < BList.size()) {
            int a = AList.get(Aidx);
            int b = BList.get(Bidx);

            // AList.get(Aidx) ==  BList.get(Bidx) 이렇게 비교하면 if에도 들어가고 else if에도 들어가는 상활 발생
            if(a == b) {
                Aidx++; Bidx++;
                Aanswer--;
                Banswer--;
            } else if(a < b) {
                Aidx++;
            } else {
                Bidx++;
            }
        }

        System.out.println(Aanswer + Banswer);

        br.close();
    }
}

문제	결과	메모리	시간	언어코드	길이
1269	맞았습니다!!	64872 KB	876 ms	Java 11 / 수정	1408

memo

• if 에서 remove 하고, else if 에서 AList.get(Aidx) 를 하면 remove 가 적용된다. 
• for 문을 한번만 쓰고 싶어서 정렬을 했는데 더 오래 걸린거 같다.

// 다른 사람 코드

import java.util.*;

public class Main
{
    public static void main(String[] args)
    {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();//A 집합의 원소 수
        int m = scanner.nextInt();//B 집합의 원소 수
        HashSet<Integer> A = new HashSet<>();//A집합
        HashSet<Integer> B = new HashSet<>();//B집합
        int result = 0;//결과


        //입력받는다
        for(int i = 0; i<n; i++)
        {
            A.add(scanner.nextInt());
        }
        for(int i = 0; i<m; i++)
        {
            B.add(scanner.nextInt());
        }

        //A집합의 원소를 돌면서 B집합에 없을 경우 result++
        for(Object i : A.toArray())
        {
            if(!B.contains(i))
            {
                result++;
            }
        }

        //B집합의 원소를 돌면서 A집합에 없을 경우 result++
        for(Object i : B.toArray())
        {
            if(!A.contains(i))
            {
                result++;
            }
        }


        System.out.println(result);
    }
}

• Set 을 활용해보고 싶었는데, iterator 는 for문에서 remove 를 쓰면 안돼서 그냥 List 사용했다.

// 다른 사람 코드

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

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		int n = Integer.parseInt(st.nextToken());
		int m = Integer.parseInt(st.nextToken());
		Set<Integer> a = new HashSet<>();
		Set<Integer> b = new HashSet<>();
		Set<Integer> c = new HashSet<>();

		st = new StringTokenizer(br.readLine());
		for (int i = 0; i < n; i++) {
			int number = Integer.parseInt(st.nextToken());
			a.add(number);
			c.add(number);
		}

		st = new StringTokenizer(br.readLine());
		for (int i = 0; i < m; i++) {
			int number = Integer.parseInt(st.nextToken());
			b.add(number);

			if (a.contains(number))
				c.remove(number);
			else
				c.add(number);
		}

		System.out.println(c.size());

	}

}