[BOJ] 10808번 - 알파벳 개수 (Swift)

문제

알파벳 소문자로만 이루어진 단어 S가 주어진다. 각 알파벳이 단어에 몇 개가 포함되어 있는지 구하는 프로그램을 작성하시오.

 

<문제 링크>

10808번: 알파벳 개수

---

입력

첫째 줄에 단어 S가 주어진다. 단어의 길이는 100을 넘지 않으며, 알파벳 소문자로만 이루어져 있다.

---

출력

단어에 포함되어 있는 a의 개수, b의 개수, …, z의 개수를 공백으로 구분해서 출력한다.

---

문제 풀이

입력된 문자열을 순회하며 해당하는 알파벳 순서에 해당하는 인덱스에 +1을 해주면 됩니다.

해당 풀이는 이중 for문을 사용하기 때문에 O(n^2) 시간 복잡도를 갖습니다.

 

하지만, 배열의 인덱스와 해당 알파벳 문자의 아스키 코드를 이용한다면 O(n) 시간 복잡도로 문제를 풀 수 있습니다.

(사실, 저는 처음 문제를 풀었을 때 이중 for문으로 풀었어요 ..ㅎ)

 

각 문자는 고유한 아스키 코드를 갖고 있으며, 연속된 알파벳들은 연속된 아스키 코드 값을 갖습니다.

 

즉, 'a'는 아스키 코드를 10진수로 표시하면 97, 'b'는 98, 'c'는 99 ... 순차적으로 이어집니다.

(추가로 'A'는 65, 'B'는 66 ... 순차적으로 아스키 코드 값을 갖습니다.) 

 

 이 문제에서는 입력이 소문자로 제한되어 있으므로 해당 사실을 활용할 수 있습니다.

풀이 순서

1. 출력 값을 저장하기 위해 알파벳 개수만큼의 요소를 갖는 배열을 초기화합니다.

2. 입력값으로 받은 문자열로 순회하며 각 문자의 아스키 코드 값을 확인합니다.

3. 해당 값을 'a'의 아스키 코드 값인 97로 빼주고, 해당 값을 배열의 인덱스로 활용하여 해당 알파벳 자리에 +1을 해줍니다.

---

풀이 코드

< 풀이 1 - 이중 for문 활용 >

 // 이중 for문 사용 풀이 
 // O(n^2)

var input = readLine()

if let input = input {
    var answer = [Int](repeating: 0, count: 26)
    var alphabets = "abcdefghijklmnopqrstuvwxyz"
    var index = 0

    for str in input {

        for alphabet in alphabets {
            if str == alphabet {
                answer[index] += 1

                index = 0
                break
            } else {
                index += 1
            }
        }
    }

    print(answer.map { String($0) }.joined(separator:" ") )
}

---

< 풀이 2 - 아스키 코드 활용 >

문자 'a'의 아스키 코드가 97인 것을 알고 있다면 97을 직접 대입하여 코드를 작성할 수 있지만,

모를 경우에는 아래와 같이 직접 확인하여 실행할 수 있습니다. 

 

(Swift에서 아스키 코드를 확인하기 위해 unicodeScalars, utf8, asciiValue 등을 활용할 수 있지만,
각 방식마다 리턴되는 타입이 다르기 때문에 주의해야합니다. 이는 나중에 Swift의 문자열을 공부하면서 더 다뤄보겠습니다.)

// 알파벳 아스키 코드를 고려하면 for문 한 번으로도 가능.
// O(n)

let input = readLine()!

var array = [Int](repeating: 0, count: 26)
var a:Character = "a"

for alphabet in input {
    let index = Int(alphabet.asciiValue! - a.asciiValue!)
    array[index] += 1
}

print(array.map{ String($0) }.joined(separator: " "))