Basic Level 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”。

  输入样例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
  

  C++

  #include <functional>
  #include <algorithm>
  #include <cstdio>
  #include <vector>
  #include <numeric>
  using namespace std;
  
  int main() {
      int N;
      int arr[10000]={0};
      scanf("%d",&N);
      for(int i=0;i<N;++i) {
          scanf("%d",&arr[i]);
      }
      vector<int> arr1,arr2,arr3,arr4,arr5;
      for(int i=0;i<N;++i){
          switch(arr[i]%5){
          case 0:
              if(arr[i]%2==0) arr1.push_back(arr[i]);
              break;
          case 1:
              arr2.push_back(arr[i]);
              break;
          case 2:
              arr3.push_back(arr[i]);
              break;
          case 3:
              arr4.push_back(arr[i]);
              break;
          case 4:
              arr5.push_back(arr[i]);
          }
      }
      //A1
      if(arr1.size()!=0){
          printf("%d ",accumulate(arr1.begin(),arr1.end(),0));
      }else
          printf("N ");
      //A2
      if(arr2.size()!=0){
          int sum = 0,sig = 1;
          for(int i=0;i<arr2.size();sig*=-1,++i)
              sum+= arr2[i]*sig;
          printf("%d ",sum);
      }else
          printf("N ");
      //A3    
      if(arr3.size()!=0){
          printf("%d ",arr3.size());
      }else
          printf("N ");
      //A4    
      if(arr4.size()!=0){
          printf("%.1f ",accumulate(arr4.begin(),arr4.end(),0)*1.0/arr4.size());
      }else
          printf("N ");
      //A%    
      if(arr5.size()!=0){
          printf("%d",*max_element(arr5.begin(),arr5.end()));
      }else
          printf("N");
        
      return 0;
  }
  
  
  

  python版

  arr = [int(x) for x in raw_input().split()][1:]
  arr1 = [x for x in arr if x%5==0 and x%2==0 ]
  arr2 = [x for x in arr if x%5==1]
  arr3 = [x for x in arr if x%5==2]
  arr4 = [x for x in arr if x%5==3]
  arr5 = [x for x in arr if x%5==4]
  
  A1 = reduce(lambda x,y:x+y,arr1,0) if len(arr1)!=0 else "N"
  A2 = (reduce(lambda x,y:x+y,[arr2[x] for x in range(0,len(arr2),2)],0) - \
       reduce(lambda x,y:x+y,[arr2[x] for x in range(1,len(arr2),2)],0)) \
       if len(arr2)!=0 else "N"
  A3 = len(arr3) if len(arr3)!=0 else "N"
  A4 = "%.1f" % (reduce(lambda x,y:x+y,arr4,0) * 1.0 / len(arr4)) if len(arr4)!=0 else "N"
  A5 = max(arr5) if len(arr5)!=0 else "N"
  print A1,A2,A3,A4,A5
  

  这里有个歧义地方：

  A3 = 被5除后余2的数字的个数；
  

  个数为0 时候应该算不存在 而不是 0