코딩테스트/프로그래머스

프로그래머스 - 가장 가까운 글자[Swift]

clamp 2023. 1. 26. 17:14

가장 가까운 같은 글자

문제 설명

문자열 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은 영어 소문자로만 이루어져 있습니다.

입출력 예sresult
"banana" [-1, -1, -1, 2, 2, 2]
"foobar" [-1, -1, 1, -1, -1, -1]

입출력 예 설명

입출력 예 #1
지문과 같습니다.

입출력 예 #2
설명 생략

 


풀이 및 소스코드

import Foundation

func solution(_ s:String) -> [Int] {
    
    var result = [Int]()
    var word_dictionary = [String: Int]()
    
    for charidx in 0..<s.count{
        let now = s.index(s.startIndex, offsetBy: charidx)
        
        //처음 나온거라면?
        if word_dictionary.contains(){ $0.key == String(s[now])} == false{
            //-1을 추가해주고
            result.append(-1)
            //단어의 인덱스를 저장해준다
            word_dictionary.updateValue(charidx, forKey: String(s[now]))
            //나온적이 있다면
        }else if word_dictionary.contains(){ $0.key == String(s[now])} == true{
            //결과 배열에 현재인덱스 - 최근 등장한 인덱스
            result.append(charidx - word_dictionary[String(s[now])]!)
            //현재 글자의 인덱스를 갱신해준다.
            word_dictionary.updateValue(charidx, forKey: String(s[now]))
        }
        
        
    }
    return result
}