https://programmers.co.kr
문제:
Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 빨간색으로 칠해져 있고 모서리는 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.
Leo는 집으로 돌아와서 아까 본 카펫의 빨간색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.
Leo가 본 카펫에서 갈색 격자의 수 brown, 빨간색 격자의 수 red가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.
풀이: 완전탐색
brown + red 인 사각형을 만족하는 모든 가로 * 세로 중에서 red를 만족하는 가로, 세로값을 찾는다.
코드: 완전탐색
사용언어 : c++
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
#include <string>
#include <vector>
using namespace std;
vector<int> solution(int brown, int red) {
vector<int> answer;
for (int i = brown/2+1;i > 0;i--) {
if ((brown + red) % i != 0)
continue;
int c = (brown + red) / i;
if ((i - 2) * (c - 2) == red) {
answer.push_back(i);
answer.push_back(c);
break;
}
}
return answer;
}
|
풀이: 수학
2*(가로) + 2*(세로) - 4 = brown
(가로) * (세로) = brown + red
를 만족하는 가로, 세로를 구한다.
코드: 수학
사용언어 : c++
1
2
3
4
5
6
7
8
9
10
11
12
|
#include <string>
#include <vector>
#include <math.h>
using namespace std;
vector<int> solution(int brown, int red) {
vector<int> answer;
double a = 0.25 * brown + 1 + sqrt(0.25 * brown * brown - 2*brown - 4*red + 4) / 2;
answer.push_back(a);
answer.push_back((brown + red) / a);
return answer;
}
|
