(Python, Java) 프로그래머스 - 카펫

19 Jan 2022 in Algorithm on Programmers, Level2

[문제 링크]

Python 풀이

import math

def solution(brown, yellow):
    tot = brown+yellow

    for height in range(1,int(math.sqrt(tot))+1):
        if tot%height == 0:
            weight = tot // height
            if ((weight-2) * (height-2)) == yellow:
                return [weight,height]

Java 풀이

class Solution {
    public int[] solution(int brown, int yellow) {
        int[] answer = new int[2];

        int tot = brown + yellow;

        for (int height = 1; height <= (int) Math.sqrt(tot); height++) {
            if (tot % height == 0) {
                int weight = tot / height;
                if (((weight - 2) * (height - 2)) == yellow) {
                    answer[0] = weight;
                    answer[1] = height;
                    break;
                }
            }
        }
        return answer;
    }
}