[Programmers] Lv2. 행렬 테두리 회전하기(kotlin)

문제

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

 

 

예시

 

 

 

풀이

• 행렬 회전 로직은 인덱스 하나씩 이동하도록 구현하면 됩니다.
• queies는 1부터 시작하니 -1을 해주시면 됩니다.

import kotlin.math.*

class Solution {
    fun solution(rows: Int, columns: Int, queries: Array<IntArray>): IntArray {
        var answer = mutableListOf<Int>()
        val arr = Array(rows){ i -> 
            IntArray(columns){ j -> i*columns + j +1}
        }
        
        queries.forEach{
            val fy = it[0] -1
            val fx = it[1] -1
            val sy = it[2] -1
            val sx = it[3] -1
            
            val last = arr[fy][fx]
            var num = last
            
            (fy..sy-1).forEach{
                arr[it][fx] = arr[it+1][fx]
                num = min(arr[it][fx], num)
            }
            
            (fx..sx-1).forEach{ 
                arr[sy][it] = arr[sy][it+1]
                num = min(arr[sy][it], num)
            }
            
            (sy downTo fy+1).forEach{
                arr[it][sx] = arr[it-1][sx] 
                num = min(arr[it][sx], num)
            }
            
            (sx downTo fx+1).forEach{
                arr[fy][it] = arr[fy][it-1] 
                num = min(arr[fy][it], num)
            }
            
            arr[fy][fx+1] = last
            answer.add(num)
        }
        
    
        return answer.toIntArray()
    }
}

 

 

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽