[프로그래머스] 카펫

 

문제 설명

 

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

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

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

 

제한사항

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

 

입출력 예

 

풀이 방법

 

우선 중점적으로 봐둔 것은 카펫을 brown, yellow로만 칠해야 하므로

brown + yellow = 가로 x 세로

두번째로, 가운데 부분을 yellow로 칠하므로

(가로-2) x (세로-2) = yellow 라는 것이다.

 

일단 brown + yellow를 통해 나오는 값은 두 수의 곱으로 이루어져야 하므로 약수의 조합으로 이루어져 있어야 한다.

따라서, ArrayList를 통해 가능한 조합만 추가해준 뒤, 위의 두 공식에 적합한 결과를 출력해주면 된다.

 

작성 코드

 

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29

import java.util.ArrayList; import java.util.Arrays;        class Solution {         public int[] solution(int brown, int yellow) {             int[] answer = new int[2];             int sum=brown+yellow;             ArrayList<Integer> list=new ArrayList<>();               for(int i=1;i<=sum;i++){                 if(sum%i==0){                     list.add(i); // 약수의 조합 (1,2,3,4,6,12), (1,2,4,8,16)                 }             }               for(int i=0;i<list.size();i++){ // 가로                 for(int j=0;j<list.size();j++){ // 세로                     if(list.get(i)*list.get(j)==sum){                         if((list.get(i)-2)*(list.get(j)-2)==yellow){                             answer[0]=list.get(i);                             answer[1]=list.get(j);                             break;                         }                     }                 }             }             return answer;         }     }