[Day12-4] 2의 영역

문제

 

내 정답 코드

import Foundation

func solution(_ arr:[Int]) -> [Int] {
    
    var index1:Int?
    var index2:Int?
    
    for i in 0..<arr.count {
        if arr[i] == 2 {
            if index1 == nil {
                index1 = i
            } else {
                index2 = i
            }
        }
    }
    
    if index1 != nil && index2 != nil {
        return Array(arr[index1!...index2!])
    } else if index1 == nil && index2 == nil {
        return [-1]
    } else {
        return [2]
    }
}

 

#

1. 반복문을 통해 arr에서 2의 값이 있는 인덱스를 저장한다.

2. 저장된 인덱스의 값에 따라 따른 값을 반환한다.

---

배운 기술