[swift] 프로그래머스 - 행렬 테두리 회전하기

프로그래머스

---

성공 풀이

func solution(_ rows:Int, _ columns:Int, _ queries:[[Int]]) -> [Int] {

    var result = [Int]()
    var arr = [[Int]]()
    
    for i in 0..<rows { // 배열 초기화
        arr.append(Array((i*columns)+1...(i*columns)+columns))
    }
    
    for i in queries { // 회전회오리!
        let startY = i[0]-1, startX = i[1]-1, endY = i[2]-1, endX = i[3]-1
        var block = Array(repeating: [Int](), count: 4) // block에 테두리를 4개로 잘라 담음
        
        for i in startX...endX { // 가로블록 구하기
            block[0].append(arr[startY][i])
            block[2].insert(arr[endY][i], at:0)
        }
        
        if startY+1 <= endY-1 { // 세로블록 구하기, 조건문이 성립이 안된다면 세로블록을 구할 필요 없음
            for i in startY+1...endY-1 {
                block[1].append(arr[i][endX])
                block[3].insert(arr[i][startX], at:0)
            }
        }
        
        var newBlock = block.flatMap {$0}
        result.append(newBlock.min()!) // 작은 블록 구하기
        
        for i in startY...endY { // 테두리를 거꾸로 돌면서 newBlock의 마지막부터 다시 넣어줌.
            arr[i][startX] = newBlock.popLast()!
        }
        for i in startX+1...endX {
            arr[endY][i] = newBlock.popLast()!
        }
        for i in (startY...endY-1).reversed() {
            arr[i][endX] = newBlock.popLast()!
        }
        if startX+1 <= endX-1 {
            for i in (startX+1...endX-1).reversed() {
            arr[startY][i] = newBlock.popLast()!
            }
        }
    }
    return result
}

풀이 키워드

- 테두리의 숫자를 순차적으로 구해 block배열에 넣어준다.
- 시계방향으로 한번 움직인 형태로 다시 배열에 block배열을 넣어준다.
- 임시배열을 만들어서 덮어씌워줬으면 반복문을 더 줄일 수 있을거 같다.

---