[LV.1] 가장 가까운 같은 글자

Heedon Ham·2024년 6월 5일
0
post-thumbnail

문자열 s가 주어졌을 때, s의 각 위치마다 자신보다 앞에 나왔으면서, 자신과 가장 가까운 곳에 있는 같은 글자가 어디 있는지 알고 싶습니다.
문자열 s이 주어질 때, 연산을 수행하는 함수 solution을 완성해주세요.

제한사항

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

입출력 예

sresult
"banana"[-1, -1, -1, 2, 2, 2]
"foobar"[-1, -1, 1, -1, -1, -1]

s ="banana"라고 할 때, 각 글자들을 왼쪽부터 오른쪽으로 읽어 나가면서 다음과 같이 진행할 수 있습니다.

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

Dictionary 활용, 해당 문자가 나타난 마지막 위치값 기억하기

각 문자와 해당 값을 -1로 초기화한 Dictionary를 정의, 문자열의 각 문자를 체크하면서 해당 값이 -1이면 처음 등장한 경우의 처리를, 다른 값이라면 가장 최근에 등장한 index를 가지고 있을 것이므로 현재 위치에서 최근 등장한 index 와의 차이를 계산한다.

import Foundation

func solution(_ s:String) -> [Int] {
    var position = ["a": -1, "b": -1, "c": -1, "d": -1, "e": -1, "f": -1, "g": -1, "h": -1, "i": -1, "j": -1, "k": -1, "l": -1, "m": -1, "n": -1, "o": -1, "p": -1, "q": -1, "r": -1, "s": -1, "t": -1, "u": -1, "v": -1, "w": -1, "x": -1, "y": -1, "z": -1]
    var result = [Int]()
    
    for (offset, element) in Array(s).enumerated() {
        if (position[String(element)] == -1) {
            result.append(-1)
        } else {
            result.append(offset - position[String(element)]!)
        }
        position[String(element)] = offset
    }
        
    return result
}
참고: 빈 Dictionary로 시작해도 무방

어차피 등장한 적 없다면 Dictionary에 존재하지 않을 것이므로 존재 여부를 먼저 체크한 뒤, 존재하면 가장 최근에 등장한 위치값을 활용해서 계산할 수 있다.

func solution(_ s: String) -> [Int] {    
    var position: [Character: Int] = [:]
    var result: [Int] = []

	s.enumerated().forEach { (offset, element) in
        if let previousOffset = position[element] {
            result += [offset - previousOffset]
            position[element] = offset
        } else {
            result += [-1]
            position.updateValue(offset, forKey: element)
        }
    }
    
    return result
}

profile
dev( iOS, React)

0개의 댓글