Algorithm

Algorithm) Programmers-보석 쇼핑 swift

행복하고 싶은 사람 2022. 11. 4. 10:02

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

Level2에서 Level3 올라오니까 난이도가 확 체감이 되네요

투포인터를 사용하는 문제였는데 이걸 못떠올려서 계~ 속 어떤 방법으로 풀까 생각하다가 찾아보고 알았습니다 ㅠ 다시 풀어봐야 겠네요

말보다는 코드 보면 쉽게 아실 수 있을것 같습니다! 

import Foundation

func solution(_ gems:[String]) -> [Int] {
    let kinds = Set(gems).count
    if kinds == 1 {
        return [1, 1]
    }
    var dic: [String: Int] = [:]
    var ans = [0, 0]
    var minDiff = Int.max
    var start = 0
    var end = 0
    
    while end < gems.count {
        if dic.count == kinds { // 모든 보석 다 있을때
            if end - start < minDiff {
                minDiff = end - start
                ans = [start + 1, end + 1]
            }
            dic[gems[start]]! -= 1
            if dic[gems[start]] == 0 {
                dic[gems[start]] = nil
            }
            start += 1
        } else { // 없을 때
            if let _ = dic[gems[end]] {
                dic[gems[end]]! += 1
                end += 1
            } else {
                dic[gems[end]] = 0
            }
        }
    }
    return ans
}