CodingTest

[코딩테스트 JAVA] 카펫

KkingKkang 2025. 2. 20. 17:31
728x90

문제 설명
Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.



Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.

Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한사항
갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.


입출력 예
brown yellow return
10        2         [4, 3]
8          1         [3, 3]
24       24        [8, 6]


접근방법

1. 모든 카펫은 1*1 로 가정한다.

2. brown 카펫과 yellow 카펫의 합은 총 카펫의 넓이이다. 

3. 카페트의 가로와 세로는 무조건  yellow 의 가로+2, 세로+2 이다. (감싸야 하니까)

4. yellow가 1개 있을 때가 최솟값인데, 그러면 최소 카페트의 가로 세로는 3*3이다. 

5. 3부터 '카페트의 총 크기-3' 까지 반복해준다. 

6. 정답을 찾으면 반복문을 빠져나온다. 

class Solution {
	public int[] solution(int brown, int yellow){
    	int[] answer = new int[2];
        int square = brown + yellow;
        int x;
        int y;
        
        for(int i=3; i<square-3; i++) {
        	if(square%i == 0){
            	x = i;
                y = square/i;
                
                if((x-2)*(y-2) == yellow){
                	answer[0] = Math.max(x,y);
                    answer[1] = Math.min(x,y);
                    break;
                }
             }
          }
          
          return answer;
     }
 }
728x90