[BOJ] 백준 26069 붙임성 좋은 총총이 (Swift)

문제

https://www.acmicpc.net/problem/26069

26069번: 붙임성 좋은 총총이

풀이

무지개 댄스는 총총이가 추고 있습니다.
무지개 댄스를 추는 사람을 만나게 되면 무지개 댄스를 추게 됩니다.

그래서 총 무지개 댄스를 추는 사람의 수를 구하는 문제인데, 어떻게 구할 수 있을까요?

저는 무지개 댄스를 추는 사람들을 담을 Set 자료형을 떠올렸습니다.
맨 처음에는 ChongChong만 담고 있습니다.

그 이후 n개의 사람들을 만난 기록을 입력 받고,
만약 무지개 댄스를 추는 Set 자료형에 있는 사람이라면
두 사람 모두 Set 자료형에 넣어줍니다.

두 사람을 모두 넣어주는 이유는 Set 자료형은 중복이 없기 때문에
그냥 코드를 짧게하기 위해서.. 두 사람 모두 넣어줬습니다.

그 이후 Set 자료형의 크기를 출력해주면 되는 문제입니다.

소스코드

	let n = Int(readLine()!)!
	var dance: Set<String> = ["ChongChong"]
	for _ in 0..<n {
	let names = readLine()!.split(separator: " ").map { String($0) }
	if dance.contains(names[0]) || dance.contains(names[1]) {
	dance.insert(names[0])
	dance.insert(names[1])
	}
	}
	print(dance.count)

view raw 붙임성 좋은 총총이.swift hosted with ❤ by GitHub

후기

Set 자료형을 떠올린다면 쉽게 풀 수 있는 문제였습니다.