PAT乙级 1054 求平均值

PAT乙级 1054 求平均值

题目

本题的基本要求非常简单：给定 N 个实数，计算它们的平均值。但复杂的是有些输入数据可能是非法的。一个“合法”的输入是 [−1000,1000] 区间内的实数，并且最多精确到小数点后 2 位。当你计算平均值的时候，不能把那些非法的数据算在内。

输入格式：

输入第一行给出正整数 N（≤100）。随后一行给出 N 个实数，数字间以一个空格分隔。

输出格式：

对每个非法输入，在一行中输出 ERROR: X is not a legal number，其中 X 是输入。最后在一行中输出结果：The average of K numbers is Y，其中 K 是合法输入的个数，Y 是它们的平均值，精确到小数点后 2 位。如果平均值无法计算，则用 Undefined 替换 Y。如果 K 为 1，则输出 The average of 1 number is Y。

输入样例 1：

7
5 -3.2 aaa 9999 2.3.4 7.123 2.35

输出样例 1：

ERROR: aaa is not a legal number
ERROR: 9999 is not a legal number
ERROR: 2.3.4 is not a legal number
ERROR: 7.123 is not a legal number
The average of 3 numbers is 1.38

输入样例 2：

2
aaa -9999

输出样例 2：

ERROR: aaa is not a legal number
ERROR: -9999 is not a legal number
The average of 0 numbers is Undefined

题解

#include <bits/stdc++.h>

bool is_real(std::string str) {
    bool flag = true;
    int point_cnt = 0, decimal_index = 0;
    bool decimal_exist = false;
    if(str[0] == '-') str = str.substr(1);
    for(int i = 0; i < str.size(); i++) {
        if(str[i] == '.') {
            decimal_exist = true;
            decimal_index = i;
            if(point_cnt == 0)
                point_cnt++;
            else {
                flag = false;
                break;
            }
        }else if(str[i] < '0' || str[i] > '9') {
            flag = false;
            break;
        }
    }
    if(!flag) return false;
    if(decimal_exist) {
        if(str.size() - 1 - decimal_index <= 2 && std::stod(str) <= 1000) return true;
        return false;
    }
    return std::abs(std::stod(str)) <= 1000;
}

int main() {
    int N;
    std::cin >> N;

    int real_num = 0;
    std::string s;
    double sum = 0;
    for (int i = 0; i < N; i++) {
        std::cin >> s;
        if(is_real(s)) {
            real_num++;
            double d = std::stod(s);
            sum += d;
        }else {
            std::cout << "ERROR: " << s << " is not a legal number" << std::endl;
        }
    }

    if(real_num == 0) std::cout << "The average of 0 numbers is Undefined" << std::endl;
    else if(real_num == 1) std::cout << "The average of 1 number is " << std::fixed << std::setprecision(2) << sum << std::endl;
    else std::cout << "The average of " << real_num << " numbers is " << std::fixed << std::setprecision(2) << sum / real_num << std::endl;

    return 0;
}