프로그래머스 - 가장 가까운 같은 글자 (Swift, Level1)

문제 👇

https://school.programmers.co.kr/learn/courses/30/lessons/142086

배열의 index와 문자열을 활용한 문제인 것 같다.

문제설명 📝

문자열 s가 주어졌을 때, s의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 알고 싶습니다.
예를 들어, s="banana"라고 할 때, 각 글자들을 왼쪽부터 오른쪽으로 읽어 나가면서 다음과 같이 진행할 수 있습니다.

b는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
a는 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
n은 처음 나왔기 때문에 자신의 앞에 같은 글자가 없습니다. 이는 -1로 표현합니다.
a는 자신보다 두 칸 앞에 a가 있습니다. 이는 2로 표현합니다.
n도 자신보다 두 칸 앞에 n이 있습니다. 이는 2로 표현합니다.
a는 자신보다 두 칸, 네 칸 앞에 a가 있습니다. 이 중 가까운 것은 두 칸 앞이고, 이는 2로 표현합니다.
따라서 최종 결과물은 [-1, -1, -1, 2, 2, 2]가 됩니다.

문자열 s이 주어질 때, 위와 같이 정의된 연산을 수행하는 함수 solution을 완성해주세요.

제한사항 💡

• 1 ≤ s의 길이 ≤ 10,000
  • s은 영어 소문자로만 이루어져 있습니다.

입출력 예

입출력 예

s	result
"banana"	[-1, -1, -1, 2, 2, 2]
"foobar"	[-1, -1, 1, -1, -1, -1]

문제풀이 ✅

1. 따로 요소들을 저장하기 위한 배열과 결과값을 저장하기 위한 배열 준비
2. 문자열을 map 함수로 하나씩 비교
3. index: 새롭게 저장하는 배열에 같은 문자가 있다면 맨 마지막의 인덱스 값
4. count: 전체 배열 길이에서 index 뺸 값

문제풀이 답

func solution(_ s:String) -> [Int] {

    var store: [Character] = []
    var result: [Int] = []

    s.map { element in
        if store.contains(element) {
            let index = store.lastIndex(of: element)!
            let count = store.count - index

            result.append(count)
        } else {
            result.append(-1)
        }
        store.append(element)
    }
    return result
}