[프로그래머스 알고리즘 고득점 Kit][완전탐색][Java] 카펫

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

프로그래머스

yellow의 약수 쌍을 이용해 노란색 영역의 가로, 세로를 구했다.
노란색 영역을 w * h라고 하면 갈색 테두리 개수는 (w + 2) * 2 + h * 2로 계산할 수 있다.
이 값이 brown과 같다면 전체 카펫의 크기는 {w + 2, h + 2}가 된다.

 

 

정답 코드

import java.util.*;

class Solution {
    public int[] solution(int brown, int yellow) {
        List<int[]> innerSquareSize = new ArrayList<>();
        for(int i=1; i*i<=yellow; i++) {

            if(yellow%i != 0) continue;
            
            innerSquareSize.add(new int[] {yellow/i, i});
        }
        
        for(int[] arr : innerSquareSize) {
            int w = arr[0];
            int h = arr[1];
            
            if((w+2) * 2 + h * 2 == brown) {
                return new int[] {w+2, h+2};
            }
        }
        
        return new int[] {-1, -1};
    }
}