프로그래머스 영어 끝말잇기 Swift

코딩테스트 연습 - 영어 끝말잇기

Foma's 풀이

중복을 확인할 변수 (for문을 1부터 돌릴것이므로 0번째 단어는 미리 넣어줍니다.) -> var w = [words[0]]

 

번호를 저장할 변수 -> var number = Int()

 

차례를 저장할 변수 ->  var time = Int()

 

for문으로 1부터 words의 갯수만큼 반복해줍니다. -> for i in 1..<words.count { ... }

 

조건 1 반복된 단어 검사

 

w안에 현재 단어(words[i])가 있는지 확인 - > if w.contains(words[i]) { ... }

 

있다면 number를 현재 단어의 인덱스 + 1 (0번째부터 시작하므로 + 1을 해줍니다.) 나누기 n의 나머지로 설정해줍니다. ->   number = (i+1)%n

 

(i+1)/n값이 딱 떨어지지 않는다면 다음 차례가 시작된것이므로 time을 (i+1)/n + 1로 아니라면 (i+1)/n으로 설정해줍니다. -> 

if Double((i+1)/n) < Double(i+1)/Double(n) {

                time = (i+1)/n + 1

            }else{

                time = (i+1)/n

            }

 

그리곤 for문을 진행할 필요가 없으므로 멈춰줍니다.  ->   break

 

아니라면 w에 단어를 넣어줍니다. - >  w.append(words[i])

 

조건 2 그 전 단어의 마지막 글자와 현재 단어의 첫번째 단어 검사

 

전 단어의 마지막 글자와 현재 단어의 첫번째 글자가 같은지 확인 - > if words[i-1].last! != words[i].first!{...}

 

조건 1과 동일하게 작성 ->  number = (i+1)%n

            if Double((i+1)/n) < Double(i+1)/Double(n) {

                time = (i+1)/n + 1

            }else{

                time = (i+1)/n

            }

            break

}

 

time이 0이라면 틀린 단어가 없다는 뜻(만약 있다면 무조건 1이상)이므로 [0,0]을 반환 - > if time == 0 {

        return [0,0]

    }

 

number가 0이라는건 (i+1)%n가 0이므로 n번째 사람이라고 볼 수 있으므로 n을 0이 아니라면 나머지를 반환해줍니다. ->

return number == 0 ? [n,time] : [number,time]

 

전체코드

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36

import Foundation   func solution(_ n:Int, _ words:[String]) -> [Int] {          var w = [words[0]]     var number = Int()     var time = Int()          for i in 1..<words.count {         if w.contains(words[i]) {             number = (i+1)%n             if Double((i+1)/n) < Double(i+1)/Double(n) {                 time = (i+1)/n + 1             }else{                 time = (i+1)/n             }             break         }         if words[i-1].last! != words[i].first!{             number = (i+1)%n             if Double((i+1)/n) < Double(i+1)/Double(n) {                 time = (i+1)/n + 1             }else{                 time = (i+1)/n             }             break         }         w.append(words[i])     }          if time == 0 {         return [0,0]     }          return number == 0 ? [n,time] : [number,time] } Colored by Color Scripter