PAT 1054 求平均值（20 分）

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

解析

这题AC率只有0.19。我现在体会到了。
你看看下面的测试数据你的程序能不能正确处理？如果不能，建议重构。

00001000.00000
00001000.00001
9.
.85
000.01

code:

#include<bits/stdc++.h>
using namespace std;
bool judge(const string& temp){
    string result  = (temp[0]=='-')?string(temp.begin()+1,temp.end()):temp;
    int count1=0,count2=0,count3=0; 
    bool flag = false,flag2=false;  
    for(auto x:result){
        if(x=='.'){
            if(flag == true)
                return false;
            flag = true;
        }
        else if(isdigit(x)){
            if(flag == false){
                if(x!='0' || flag2==true){
                    count1++;
                    flag2 = true;
                }
            }
            else{
                count2=(x=='0')?count2+1:0;
                count3++;
            }
        }
        else
            return false;
    }
    count2=count3-count2;
    if((count1==4 && count2==0) ||(count1<=3 && count2<=2)) 
        return true;
    else
        return false;
}
int main()
{
  int N,count=0;
  string str;
  double sum=0.0;
  cin>>N;
  while(N--){
    cin>>str;
    if(judge(str)){
        double temp = stod(str);
        if(temp>=-1000 && temp<=1000){
            sum+=temp;
            count++;
        }
        else
            cout<<"ERROR: "<<str<<" is not a legal number"<<endl;
    }
    else
      cout<<"ERROR: "<<str<<" is not a legal number"<<endl;
  }
  if(count == 0)
    cout<<"The average of 0 numbers is Undefined";
  else if(count == 1)
    printf("The average of 1 number is %.2lf",sum);
  else
    printf("The average of %d numbers is %.2lf",count,sum/count);
}