PAT 数字分类

1012 数字分类 (20分)

给定一系列正整数，请按要求对数字进行分类，并输出以下 5 个数字：
A​1 = 能被 5 整除的数字中所有偶数的和；
A​2 = 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算 n​1−n2+n​3−n4⋯；
A​3 = 被 5 除后余 2 的数字的个数；
A4= 被 5 除后余 3 的数字的平均数，精确到小数点后 1 位；
A​5​​= 被 5 除后余 4 的数字中最大数字。
输入格式：
每个输入包含 1 个测试用例。每个测试用例先给出一个不超过 1000 的正整数 N，随后给出 N 个不超过 1000 的待分类的正整数。数字间以空格分隔。
输出格式：
对给定的 N 个正整数，按题目要求计算 A​1​​ ~A​5
并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。
若其中某一类数字不存在，则在相应位置输出 N。

输入样例 1：
13 1 2 3 4 5 6 7 8 9 10 20 16 18
输出样例 1：
30 11 2 9.7 9
输入样例 2：
8 1 2 4 5 6 7 9 16
输出样例 2：
N 11 2 N 9

#include <iostream>
#include <iomanip>
using namespace std;

int main(){
    int ans[5] = {0}; //以后做题 该初始化为0的就初始化为0，以免发生一些未知的错误
    int count[5] = {0};
    
    int n;
    cin >> n;
    int a, cnt = 0, max = 0; //无论是数组还是变量 一定要记得初始化！！！
    
    bool flag = true;
    while(n--){
        cin >> a;
        if(a % 2 == 0 && a % 5 == 0){
            count[0] = 1;
            ans[0] += a;
        }
        
        if(a % 5 == 1){
            count[1] = 1;
            if(flag){ //true加false减
                flag = false;
                ans[1] += a;
            } 
            else{
                flag = true;
                ans[1] -= a;
            } 
        }
        
        if(a % 5 == 2){
            count[2] = 1;
            ans[2]++;
        }
        
        if(a % 5 == 3){
            count[3] = 1;
            ans[3] += a;
            cnt++;  
        }
        
        if(a % 5 == 4){
            count[4] = 1;
            if(a > max){
                max = a;
            }
            ans[4] = max;
        }
    }
    
    for(int i=0; i<4; i++){
        if(i == 3 && count[i]){
            cout << fixed << setprecision(1) << (double)ans[i] / cnt << " ";
        }else if(count[i]){
            cout << ans[i] << " ";
        }else {
        	cout << "N" << " ";
		}
    }
    //最后一个的输出不能有空格
    if(count[4])
    	cout << ans[4] << endl; 
	else 
		cout << "N" << endl;
    return 0;
}