PAT乙级练习题B1054. 求平均值

题目描述

本题的基本要求非常简单：给定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<iostream>
#include<string>
#include<cmath>
using namespace std;

bool is_good(const string &str)
{
  int point = 0,frac = 0;
  for (int i = 0; i < str.size(); ++i)
  {
    if (i == 0&&str[i]=='-'&&str.size()>1)
    {
      continue;
    }
    if (str[i] <= '9'&&str[i] >= '0')
    {
      if (point == 1)
      {
        ++frac;
        if (frac > 2)
        {
          return false;
        }
      }
    }
    else
    {
      if (str[i] == '.')
      {
        ++point;
        if (point > 1)
        {
          return false;
        }
      }
      else
      {
        return false;
      }
    }
  }
  double val = atof(str.c_str());
  if (val > 1000 || val < -1000)
  {
    return false;
  }
  return true;
}

int main()
{
  int N,cnt=0;
  double val, sum = 0;
  string str;
  cin >> N;
  for (int i = 0; i < N; ++i)
  {
    cin >> str;
    if (is_good(str))
    {
      val = atof(str.c_str());
      sum += val;
      ++cnt;
    }
    else
    {
      printf("ERROR: %s is not a legal number\n", str.c_str());
    }
  }
  if (cnt == 0)
  {
    cout << "The average of 0 numbers is Undefined" << endl;
  }
  else
  {
    if (cnt == 1)
    {
      printf("The average of 1 number is %.2f", sum );
    }
    else
    {
      printf("The average of %d numbers is %.2f", cnt, sum / cnt);
    }
  }
  system("pause");
  return 0;
}