1054. 求平均值 (20)

本题的基本要求非常简单：给定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>
using namespace std;
bool islegal(string &s){
    int inpcnt,dotcnt=0,deccnt=0;
    for(size_t i=0;i<s.size();++i){
        if(i==0&&s[i]=='-') continue;
        if(isdigit(s[i])&&dotcnt==0){
            ++inpcnt;continue;
        }
        if(isdigit(s[i])&&dotcnt==1&&deccnt<2){
            ++deccnt;continue;
        }
        if(s[i]=='.'&&inpcnt>0&&dotcnt==0){
            ++dotcnt;continue;
        }
        return false;
    }
    return true;
}
int main(){
    double sum=0.0;
    int n,cnt=0;
    cin>>n;
    while(n--){
        string str;
        cin>>str;
        if(islegal(str)){
            double num=stod(str);
            if(num>=-1000&&num<=1000){
                sum+=num;
                cnt++;
            }
            else printf("ERROR: %s is not a legal number\n",str.c_str());
        }
        else printf("ERROR: %s is not a legal number\n",str.c_str());
    }
    if(cnt>1) printf("The average of %d numbers is %.2lf\n",cnt,sum/cnt);
    else if(cnt==1) printf("The average of 1 number is %.2lf\n",sum);
    else printf("The average of 0 numbers is Undefined\n");
}