[Programmers] Lv2.카펫(kotlin)

문제

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

 

 

예시

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

 

 

 

풀이

• 완전탐색 방식으로 yellow 길이의 루트만큼만 반복해서 가로, 세로 길이를 구합니다. 해당 길이에 각각 +2씩 더한 후 사각형의 크기를 구하고 brown과 yellow의 갯수의 합과 같은지 확인하면 됩니다.

import kotlin.math.*

class Solution {
    fun solution(brown: Int, yellow: Int): IntArray {
        (1..sqrt(yellow.toDouble()).toInt()).forEach{
            if(yellow % it == 0){
                val x = it + 2 
                val y = yellow / it + 2
                
                if(x*y == brown + yellow){
                    return intArrayOf(y, x)
                }
            }
        }
        return intArrayOf()
    }
}

 

 

 

참고

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