[백준] 1484번 : 다이어트 (C++)

1484번 : 다이어트

---

문제)

성원이는 다이어트를 시도중이다. 성원이는 정말 정말 무겁기 때문에, 저울이 부셔졌다. 성원이의 힘겨운 다이어트 시도를 보고만 있던 엔토피아는 성원이에게 새로운 저울을 선물해 주었다. 성원이는 엔토피아가 선물해준 저울 위에 올라갔다. “안돼!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! G 킬로그램이나 더 쪘어ㅜㅠ”라고 성원이가 말했다. 여기서 말하는 G킬로그램은 성원이의 현재 몸무게의 제곱에서 성원이가 기억하고 있던 몸무게의 제곱을 뺀 것이다.

성원이의 현재 몸무게로 가능한 것을 모두 출력하는 프로그램을 작성하시오.

 

 

 

입력 :

첫째 줄에 G가 주어진다. G는 100,000보다 작거나 같은 자연수이다.

 

 

 

출력 :

첫째 줄부터 한 줄에 하나씩 가능한 성원이의 현재 몸무게를 오름차순으로 출력한다. 가능한 몸무게가 없을 때는 -1을 출력한다. 현재 몸무게는 자연수로 떨어지지 않을 수도 있는데, 이런 경우는 제외해야 한다.

 

 

 

 

 

 

 

풀이)

// 1484. 다이어트
#include <iostream>
#include <vector>
#include <math.h>
 
using namespace std;
 
vector<int> answer;
 
int main()
{
    int g;
    cin >> g;
 
    vector<int> vec;
    for (int i = 0; i < g + 1; i++) vec.push_back(i);
 
    // 몸무게는 0일 수 없다.
    int s = 1;
    int e = 1;
 
    // 성원이의 현재 몸무게가 g에 도달할 때까지
    while (e <= g)
    {
        // 성원이가 기억하던 몸무게
        long long _a = pow(s, 2);
 
        // 성원이의 현재 몸무게
        long long _b = pow(e, 2);
 
        // G 킬로그램과 같다면
        if (_b - _a == g)
        {
            answer.push_back(e);
            s++;
        }
        else if (_b - _a < g) e++;
        else if (_b - _a > g) s++;
    }
 
 
    if (answer.size() == 0) cout << -1;
    else for (int ele : answer) cout << ele << '\n';
 
    return 0;
}
 

---

출처 : https://www.acmicpc.net/problem/1484 

1484번: 다이어트