[프로그래머스] 카펫

문제설명

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

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

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

제한사항

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

해결 방법

갈색으로 배열의 테두리를 만들었을 때, 테두리를 제외한 원소의 개수 빨간색과 일치하는 경우를 찾는다.

 

JAVA

package programmers;

import java.util.*;

public class p_42842 {
	public static int[] solution(int b, int r) {
		int[] answer = new int[2];
		for(int i = 3; i <= (b + r) / 2; i++) {
			if((b + r) % i == 0) {
				if((i - 2) * ((b + r ) / i - 2) == r) {
					answer[0] = (b + r) / i;
					answer[1] = i;
					break;
				}
			}
		}
		return answer;
	}
	public static void main(String[] args) {
		int[] r = solution(10, 2);
		for(int t : r) {
			System.out.println(t);
		}
	}
}