[프로그래머스][자바] 카펫

문제 설명

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

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

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

제한사항

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

---

 

 

🙄 생각해낸 방법

- 가로 * 세로 = brown + red

- 가로-2 * 세로-2 = red

- 제곱근이 보통 값으로 나온다

- 제곱근이 아니면 그 근처다

 

 

이렇게 푸는게 맞는건가;; 다른사람들 코드도 봐봐야겠다.. 

 

---

public static int[] solution(int brown, int red) {
	        int[] answer = new int[2];
	        
	        int sum = brown + red;  
	        
	        //제곱근 구하기
	        double sqrt = Math.sqrt(sum);
	        int a = (int)sqrt;
	        int b = sum/(int)sqrt;

			//제곱근이 답이 아니면 제곱근 근처에서 찾기
	        //a*b가  sum 값이랑 같고
	        //a-2 * a-b 곱한 값이 red 값이랑 같아야함 
	        if((a*b != sum) || ((a-2)*(b-2) != red)) {
	        	while(a>=1) {
	        		a = a-1;
	        		if(sum % a ==0) {
	        			b = sum/a;
	        			if((a*b == sum) && ((a-2)*(b-2) == red)) {
		        			break;
		        		}
	        		}
	        		
	        	}
	        }
	        // a b 순서대로 넣어주기 
	        if (a>=b) {
	        	answer[0]=a;
	        	answer[1]=b;
	        }else {	
	        	answer[0]=b;
	        	answer[1]=a;
	        }
	        
	        return answer;
	}

---

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

프로그래머스