[백준/C++] 19636번: 요요 시뮬레이션

문제

데시는 D일 동안 다이어트를 하기로 결심했다.

다이어트 전 데시의 체중은 W0 g, 일일 에너지 섭취량 I0 Kcal, 일일 기초 대사량은 일일 에너지 섭취량과 같은 I0 Kcal이다. 데시는 운동을 하지 않아 일일 활동 대사량 A0은 0 Kcal이다.

다이어트 기간, 데시는 일일 에너지 섭취량과 일일 활동 대사량을 바꿀 것이다(물론 다이어트 전과 같을 수도 있다). 데시의 다이어트 기간 일일 에너지 섭취량은 I Kcal, 다이어트 기간 일일 활동 대사량은 A Kcal이다.

다이어트를 하는 동안 데시의 체중과 일일 기초 대사량은 변화한다. 단, 데시의 신체 구조는 매우 단순하여 체중과 일일 기초 대사량은 각각 하루에 단 한 번만 변화한다.

체중은 (일일 에너지 섭취량 − 일일 에너지 소비량) g/Kcal만큼 더해진다고 가정한다. 일일 에너지 소비량은 일일 기초 대사량과 일일 활동 대사량을 더한 값이다.

(일일 에너지 섭취량 − 일일 에너지 소비량) 의 절댓값이 기초 대사량 변화 역치 T Kcal 초과라면, 일일 기초 대사량은 ⌊ (일일 에너지 섭취량 − 일일 에너지 소비량) / 2 ⌋ 만큼 더해진다. 일일 기초 대사량 변화는 같은 날의 체중 변화 다음에 일어난다.

단, 아래에 해당하면 데시는 사망한다.

• 체중이 0 g 이하인 경우
• 일일 기초 대사량이 0 Kcal 이하인 경우

D일 간의 다이어트가 끝난 후, 일일 기초 대사량의 변화를 고려하지 않았을 때와 일일 기초 대사량의 변화를 고려했을 때 각각의 예상 체중과 일일 기초 대사량을, 그리고 다이어트 전 데시의 원래 생활로 돌아간다면 몸무게가 증가하는 요요 현상이 일어날지 계산해주자.

입력

첫 번째 줄에는 데시의 다이어트 전 체중 W0 (1 ≤ W0 ≤ 107), 다이어트 전 일일 에너지 섭취량 및 일일 기초 대사량 I0 (1≤ I0 ≤ 105), 기초 대사량 변화 역치 T (1 ≤ T ≤ 1,000)가 빈칸을 사이에 두고 주어진다.

두 번째 줄에는 데시의 다이어트 기간 D (1 ≤ D ≤ 1,000), 다이어트 기간 일일 에너지 섭취량 I (0 ≤ I ≤ 105), 다이어트 기간 일일 활동 대사량 A (0 ≤ A ≤ 105)가 빈칸을 사이에 두고 주어진다.

입력으로 주어지는 모든 수는 정수이며, 단위는 지문에서 언급한 단위를 따른다. 다이어트 동안 체중이 오히려 증가하는 경우는 입력으로 주어지지 않는다.

출력

첫 번째 줄에는 일일 기초 대사량의 변화를 고려하지 않았을 때의 다이어트 후 예상 체중과 일일 기초 대사량을 출력한다. 일일 기초 대사량 변화를 고려하지 않았으므로 다이어트 전 일일 기초 대사량을 그대로 출력하면 된다. 만약 다이어트 도중 데시의 사망이 예상되는 경우 체중과 일일 기초 대사량 대신 Danger Diet를 출력한다.

두 번째 줄에는 일일 기초 대사량의 변화를 고려했을 때의 다이어트 후 예상 체중과 일일 기초 대사량을 출력하고, 변화한 일일 기초 대사량을 가지고 다이어트 전 일일 에너지 섭취량과 다이어트 전 일일 활동 대사량으로 돌아갔을 때 체중이 증가하는 요요 현상이 발생하는 경우 YOYO, 발생하지 않는 경우 NO를 출력한다. 만약 다이어트 도중 데시의 사망이 예상되는 경우 체중과 일일 기초 대사량, 요요 발생 여부 대신 Danger Diet를 출력한다.

 

풀이

단순 구현이긴 하지만 문제가 길어서 헷갈린다.^^ 데시야 그냥 아무 생각없이 운동하면서 다이어트하면 안 되는 거니

출력해야 할 것

1) 일일 기초 대사량 변화를 고려하지 않았을 때 : 예상 체중과 일일 기초 대사량

- 일일 기초 대사량이 변화하지 않으므로 주어진 수식을 사용하면 된다.

int W = W0 + (I - (I0 + A)) * D;

- 일일 기초 대사량(I0)은 0이 아닌 것으로 주어졌으니 체중만 확인해서 사망 여부를 출력한다.

2) 일일 기초 대사량 변화를 고려했을 때 : 예상 체중과 일일 기초 대사량

다이어트 기간 동안 아래 공식을 적용시켜서 체중을 감량한다.

체중 += 일일 에너지 섭취량(I) - 일일 에너지 소비량(O) (O = 일일 기초 대사량(I0) + 일일 활동 대사량(A)

이때 변화량이 역치 T를 초과하면 기초 대사량을 변화시켜준다.

if (abs(I-O) > T)
    I0 += floor((double)(I - O)/2);

매일 일일 기초 대사량과 체중을 확인하여 사망했을 경우 바로 리턴한다. 

if (W0 <= 0 || I0 <= 0)
    return ci(0, 0);

 

3) 요요 여부

요요의 조건 : 변화한 일일 기초 대사량을 가지고 다이어트 전 일일 에너지 섭취량(I0)과 활동 대사량(0)으로 돌아갈 경우 체중 증가 여부를 따지면 된다.

 

전체 코드

#include <iostream>
#include <cmath>
using namespace std;
typedef pair<int, int> ci;

ci diet(int D, int W0, int I, int I0, int T, int A){
    while(D--){
        int O = I0 + A; // 일일 에너지 소비량
        W0 += (I - O); // 체중

        if (abs(I-O) > T)
            I0 += floor((double)(I - O)/2);

        if (W0 <= 0 || I0 <= 0)
            return ci(0, 0);
    }
    return ci(W0, I0);
}

int main(){

    /*
     * 체중 : 일일 에너지 섭취량(I) - 일일 에너지 소비량(O) (일일 기초 대사량(I0) + 일일 활동 대사량(A)
     * IF |일일 에너지 섭취량(I) - 일일 에너지 소비량(O)| > T
     * -> 일일 기초 대사량(B) += floor(일일 에너지 섭취량(I) - 일일 에너지 소비량(O)/2)
     *
     * 사망 조건 : 체중(W) <= 0 || 일일 기초 대사량(I0) <= 0
     */

    int W0, I0, T, D, I, A;
    cin >> W0 >> I0 >> T; // 다이어트 전 체중(W0), 일일 에너지 섭취량(I0), 일일 기초 대사량(I0), 기초대사량 역치(T)
    cin >> D >> I >> A; // 다이어트 기간(D), 일일 에너지 섭취량(I), 일일 활동 대사량(A)

    // 1) 일일 기초 대사량의 변화 고려 X
    int W = W0 + (I - (I0 + A)) * D;
    if (W <= 0)
        cout << "Danger Diet\n";
    else
        cout << W << " " << I0 << "\n";

    // 2) 일일 기초 대사량의 변화 고려 O
    ci result = diet(D, W0, I, I0, T, A);
    if (result.first <= 0 || result.second <= 0)
        cout << "Danger Diet";

    // 3) 요요 여부
    else {
        string ans = "NO";
        // 변화한 일일 대사량을 가지고 다이어트 전 일일 에너지 섭취량(I0)과 활동 대사량(0)으로 돌아갈 경우
        if (I0 - result.second > 0)
            ans = "YOYO";
        cout << result.first << " " << result.second << " " << ans << "\n";
    }
}