[Programmers] Lv1.N^2 배열 자르기(Kotlin)

• 이 문제는 kotlin으로 풀이를 작성하였습니다.

문제

https://programmers.co.kr/learn/courses/30/lessons/87390

 

예시

https://programmers.co.kr/learn/courses/30/lessons/87390

 

풀이

• 배열을 구한 후에 답을 구하면 쉽게 구할 수 있지만 시간초과를 피할 수 없습니다.
• 해당 문제는 배열을 굳이 구할 필요없이 규칙을 잘 파악하면 원하는 지점부터 값을 구할 수 있습니다.
• 1번째 행에서는 1번째 열까지 1로 채우고 이후부터는 1씩 증가합니다.
  2번째 행에서는 2번째 열까지 2로 채운 후 1씩 증가합니다.
  ...
  n번째 행에서는 n번째 열까지 n으로 채웁니다.

class Solution {
    fun solution(n: Int, left: Long, right: Long): IntArray {
        var answer = mutableListOf<Long>()
        
        var start_row = left/n; var start_col = left%n
        var end_row = right/n; var end_col = right%n;
        
        if(start_row == end_row){
            for(i in start_col..end_col){
                if(i<=start_row) answer.add(start_row + 1)
                else answer.add(i+1)
            }
        }else{
            for(i in start_col until n){
                if(i<=start_row) answer.add(start_row + 1)
                else answer.add(i+1)
            }
            
            for(i in start_row+1 until end_row){
                for(j in 0 until n.toLong()){
                    if(j<=i) answer.add(i+1)
                    else answer.add(j+1)
                }
            }
            
            for(i in 0..end_col){
                if(i<=end_row) answer.add(end_row+1)
                else answer.add(i+1)
            }
        }
        
        return answer.map { it.toInt()}.toIntArray();
    }
}

 

• 위에 규칙을 잘 생각하면 index/n 과 index%n 중에서 최댓값이 배열의 값이 된다는 것을 알 수 있습니다.
• 해당 문제 풀이 시 intArray와 ArrayList<Int> 를 잘 구별하여 사용하면 편리하게 풀 수 있습니다.

import kotlin.math.max

class Solution {
    fun solution(n: Int, left: Long, right: Long): IntArray {
        return (left..right).map{ (max(it/n, it%n)+1).toInt() }.toIntArray()
    }
}