PAT乙级练习题B1012. 数字分类

问题描述

给定一系列正整数，请按要求对数字进行分类，并输出以下5个数字：

A1 = 能被5整除的数字中所有偶数的和；
A2 = 将被5除后余1的数字按给出顺序进行交错求和，即计算n1-n2+n3-n4…；
A3 = 被5除后余2的数字的个数；
A4 = 被5除后余3的数字的平均数，精确到小数点后1位；
A5 = 被5除后余4的数字中最大数字。
输入格式：

每个输入包含1个测试用例。每个测试用例先给出一个不超过1000的正整数N，随后给出N个不超过1000的待分类的正整数。数字间以空格分隔。

输出格式：

对给定的N个正整数，按题目要求计算A1~A5并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。

若其中某一类数字不存在，则在相应位置输出“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

题目解析

使用vector，将满足条件的数字填入vector后按要求输出。
看清题目，注意输入的格式，第一个数字是数字个数。

代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

void getA(vector<int>&A1, vector<int>&A2, vector<int>&A3, vector<int>&A4, vector<int>&A5, vector<int>&list)
{
    for (int i = 0; i < list.size(); ++i)
    {
        int num = list[i];
        if (num % 10 == 0)
        {
            A1.push_back(num);
        }
        switch (num % 5)
        {
        case 1:
            A2.push_back(num);
            break;
        case 2:
            A3.push_back(num);
            break;
        case 3:
            A4.push_back(num);
            break;
        case 4:
            A5.push_back(num);
            break;
        }
    }

}



int main()
{
    int N;
    vector<int> list;
    vector<int> A1;
    vector<int> A2;
    vector<int> A3;
    vector<int> A4;
    vector<int> A5;
    cin>>N;
    for (int i = 0; i < N; ++i)
    {
        int val;
        cin >> val;
        list.push_back(val);
    }
    getA(A1,A2,A3,A4,A5,list);
    if (!A1.empty())
    {
        int A1_sum = 0;
        for (auto it = A1.begin(); it !=A1.end(); ++it)
            A1_sum += *it;
        cout << A1_sum << " ";
    }
    else
    {
        cout << "N ";
    }

    if (!A2.empty())
    { 
        int A2_sum=0;
        for (int i = 0; i < A2.size(); ++i)
        {
            if (i % 2 == 0)
            {
                A2_sum += A2[i];
            }
            else
            {
                A2_sum -= A2[i];
            }
        }
        cout << A2_sum << " ";
    }
    else
    {
        cout << "N ";
    }

    if (!A3.empty())
    {
        cout << A3.size() << " ";
    }
    else
    {
        cout << "N ";
    }

    if (!A4.empty())
    {
        float A4_sum = 0;
        for (int i = 0; i < A4.size(); ++i)
        {
            A4_sum += A4[i];
        }
        float aver=A4_sum/A4.size();
        printf("%.1f", aver);
        cout << " ";
    }
    else 
    {
        cout << "N ";
    }

    if (!A5.empty())
    {
        cout << *max_element(A5.begin(), A5.end()) << endl;
    }
    else
    {
        cout << "N" << endl;
    }
    system("pause");
    return 0;

}