문제 설명
Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 빨간색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.
Leo는 집으로 돌아와서 아까 본 카펫의 빨간색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.
Leo가 본 카펫에서 갈색 격자의 수 brown, 빨간색 격자의 수 red가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.
제한사항
- 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
- 빨간색 격자의 수 red는 1 이상 2,000,000 이하인 자연수입니다.
- 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.
입출력 예
brown | red | return |
10 | 2 | [4, 3] |
8 | 1 | [3, 3] |
24 | 24 | [8, 6] |
문제 풀이
금방 풀었다.
for문을 돌리면서 3가지만 주의하면 됐다.
1. 가로의 길이가 세로의 길이와 같거나 더 길어야 한다.
2. 찾은 가로, 세로 길이를 통해 넓이를 확인해보자.
3. 그리고 brown의 테두리가 1줄만 있어야한다.
Solution.cpp
#include <string>
#include <vector>
using namespace std;
vector<int> solution(int brown, int red) {
vector<int> answer;
for(int i=3; i<brown/2; i++) { // 가로
for(int j=3; j<=i; j++) { // 세로
// 넓이가 같아야하고, brown이 테두리 1줄만 있어야 함
if((i*j==brown+red) && (2*i+(2*j-4))==brown) {
answer.push_back(i);
answer.push_back(j);
return answer;
}
}
}
return answer;
}
문제 출처: https://programmers.co.kr/learn/courses/30/lessons/42842#
'알고리즘 문제풀이 > 문제풀이' 카테고리의 다른 글
[c++] 프로그래머스: 스택/큐 / 기능개발 (0) | 2020.04.02 |
---|---|
[c++] 백준: 재귀 / 팩토리얼 (0) | 2020.04.01 |
[c++] 프로그래머스: 완전탐색 / 숫자 야구 (0) | 2020.03.28 |
[c++] 프로그래머스: 완전탐색 / 소수 찾기 (0) | 2020.03.27 |
[c++] 프로그래머스: 깊이 우선 탐색(DFS) / 단어 변환 (0) | 2020.03.21 |