백준 알고리즘

[Silver IV] 대칭 차집합 - 1269

성능 요약

메모리: 25556 KB, 시간: 60 ms

분류

자료 구조, 해시를 사용한 집합과 맵, 트리를 사용한 집합과 맵

문제 설명

자연수를 원소로 갖는 공집합이 아닌 두 집합 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을 넘지 않는다.

출력

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

Code

use std::collections::HashSet;
use std::io::{stdin, BufRead, BufReader, BufWriter, Write};

fn main() {
    let stdin = stdin();
    let mut reader = BufReader::new(stdin.lock());

    let mut buffer = String::new();
    reader.read_line(&mut buffer).unwrap();
    let mut nums = buffer.trim().split_whitespace();
    let a_len = nums.next().unwrap().parse::<usize>().unwrap();
    let b_len = nums.next().unwrap().parse::<usize>().unwrap();

    let mut a_set = HashSet::new();
    let mut buffer_a = String::new();
    reader.read_line(&mut buffer_a).unwrap();
    let mut nums_a = buffer_a.trim().split_whitespace();
    for _ in 0..a_len {
        let a_val = nums_a.next().unwrap().parse::<usize>().unwrap();
        a_set.insert(a_val);
    }

    let mut b_set = HashSet::new();
    let mut buffer_b = String::new();
    reader.read_line(&mut buffer_b).unwrap();
    let mut nums_b = buffer_b.trim().split_whitespace();
    for _ in 0..b_len {
        let b_val = nums_b.next().unwrap().parse::<usize>().unwrap();
        b_set.insert(b_val);
    }

    let a_diff = a_set.difference(&b_set).count();
    let b_diff = b_set.difference(&a_set).count();
    let stdout = std::io::stdout();
    let mut writer = BufWriter::new(stdout.lock());

    writeln!(writer, "{}", a_diff + b_diff).unwrap();
}

백준 1269 - 대칭 차집합[Rust]