【洛谷 · for C++】P1422 小玉家的电费

题目链接

题目本身非常简单，直接if-else其实就能做，我的代码用了一个表驱动。
需要注意的是（总是会忘记的一个点，也导致这次没能一次AC的一个点）：
这是一种分段计算的题目，每一个符合的段都应该分开计算然后最后加起来，而不是只要计算最大符合的一段！

贴一下自己的代码（刚开始学C++，代码可能写得比较xxx）：

#include <iostream>
#include <iomanip>

using namespace std;

int main() {
    int amount[3] = { 400, 150, 0 };
    double price[3] = { 0.5663, 0.4663, 0.4463 };

    int sum;
    cin >> sum;

    double res = 0.0;

    for (int i = 0; i < 3; i++) {
        if (sum > amount[i]) {
            res += (sum - amount[i]) * price[i];
            sum = amount[i];
        }
    }

    cout << fixed << setprecision(1);
    cout << res;

    return 0;

}