PTA_B_1054 求平均值 （20 分）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<algorithm>
#include<map>
using namespace std;

double num = 0;

int legal(char s[]){
    double num1 = 0, num2 = 0;
    int n = strlen(s);
    int neg = 0;
    int p = n;
    for(int i = 0; i < n; i++){
        if(s[i] == '.'){
            p = i; break;
        }
    }
    //cout << "p: " << p << " ";
    int p1 = 0, p2 = n-1;
    if(s[0] == '-'){neg = 1; p1 = 1;}
    while(p1 < p){
        if(!isdigit(s[p1])) return 0;
        num1 = num1*10 + s[p1++] - '0';
    }
    if(p2 - p > 2) return 0;
    while(p2 > p){
        if(!isdigit(s[p2])) return 0;
        num2 = num2/10 + s[p2--] - '0';
    }
    num2 /= 10;
    num = num1 + num2;
    if(num)
    if(neg) num *= -1;
    if(num > 1000.0 || num < -1000.0) return 0;
    //cout << "num: " << num << endl;
    return 1;
}
int main(){
    int n, cnt = 0;;
    scanf("%d", &n);
    char s[110];
    double sum = 0;
    for(int i = 0; i < n; i++){
        scanf("%s", s);
        if(legal(s)){cnt++; sum += num;}
        else printf("ERROR: %s is not a legal number\n", s);
    }
    if(cnt == 0) printf("The average of 0 numbers is Undefined\n");
    else if(cnt == 1) printf("The average of 1 number is %.2lf\n",sum);
    else printf("The average of %d numbers is %.2f\n", cnt, sum/cnt);

}