PAT乙级-1012

0. 题目

1. 解析

新建一个大小为5的数组，元素为vector，将输入的数分别按余数存储至不同的vector中，最后根据不同特征来分别计算A1~A5.

2. 代码

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

int main(int argc, char **argv)
{
    vector<int> m[5];
    int n,inp,A1=0,A2=0,A3=0,A5=0;
    float A4_sum=0,A4_cnt=0;
    cin>>n;
    while(n--){
        cin>>inp;
        m[inp%5].push_back(inp);
    }

    for(int i=0;i<5;i++){
        for(int j=0;j<m[i].size();j++){
            if(i==0 && m[i][j]%2==0) A1+=m[i][j];
            if(i==1 && j%2==0) A2+=m[i][j];
            if(i==1 && j%2==1) A2-=m[i][j];
            if(i==2) A3++;
            if(i==3) A4_sum+=m[i][j],A4_cnt++;
            if(i==4 && m[i][j]>A5) A5=m[i][j];
        }
    }

    for(int i=0;i<5;i++){
        if(i!=0) printf(" ");
        //输出A1但A1==0，或者输出Ai但其vector中没有数，证明该分类不存在数字，故输出“N”
        if(i==0 && A1==0 || i!=0 && m[i].size()==0){printf("N");continue;} 
        if(i==0) printf("%d",A1);
        if(i==1) printf("%d",A2);
        if(i==2) printf("%d",A3);
        if(i==3) printf("%.1f",A4_sum/A4_cnt);
        if(i==4) printf("%d",A5);

    }
    return 0;
}