C++/Python PAT乙级 1012 数字分类 (20)

题目描述

给定一系列正整数，请按要求对数字进行分类，并输出以下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”。

输入例子:

13 1 2 3 4 5 6 7 8 9 10 20 16 18

输出例子:

30 11 2 9.7 9

C++

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

int main()
 {
     int N,A1=0,A2=0,A3=0,A5=0;
     cin >> N ;
     int t1=0,t2=0,t3=0,t4=0,t5=0;
     float A4,sum_A4=0;
     int num[N];
     for (int i=0;i<=N-1;i++)
     {
         cin >> num[i];
     }
     //判断
     for (int i=0 ;i<=N-1;i++)
     {
         //A1
         switch (num[i]%5)
         {
             case 0:
                   if ( num[i]%2  ==0)
                   {
                   t1++;
                   A1+=num[i];
                   }break;
         //A2
             case 1:
                  {
                   A2 += num[i]*pow(-1,t2);
                   t2++;
                   break;
                  }
         //A3
             case 2:
                   t3++;
                   A3=t3;
                   break;

         //A4
            case 3:
                   t4++;
                   sum_A4 +=num[i];
                   A4 = sum_A4/t4;
                   break;
         //A5
            case 4:
                   t5++;
                   if ( A5 < num[i])
                       {A5 = num[i];}
                   break;


     }
     }
//cout << A4 <<endl;
     if(t1==0)
          cout <<'N'<<' ';
     else cout <<A1<<' ';

     if(t2==0)
          cout <<'N'<<' ';
     else cout <<A2<<' ';

     if(t3==0)
          cout <<'N'<<' ';
     else cout <<A3<<' ';

     if(t4==0)
          cout <<'N'<<' ';
     else cout << setiosflags(ios::fixed)<<setprecision(1)<<A4<<' ';

     if(t5==0)
          cout <<'N';
     else cout <<A5;

     return 0;

 }

Python

n=list(input().split())
a=[0,0,0,0,0]
count=[0,0,0,0,0]
for i in range(1,len(n)):
    t=int(n[i])
    count[t%5]+=1
    if t%5==0:
        if t%2==0:
            a[0]+=t
        else:
            count[0]-=1
    elif t%5==1:
        a[1]+=t*((-1)**(count[1]-1))
    elif t%5==2:
        a[2]+=1
    elif t%5==3:
        a[3]+=t
    elif t%5==4:
        if t>a[4]:
            a[4]=t
for i in range(5):
    if count[i]==0:
        a[i]='N'
if count[3]!=0:
    a[3]=round(a[3]/count[3],1)
print(" ".join(str(a[i]) for i in range(5)))