PAT乙级 1012. 数字分类

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

  代码如下（python）：

  __author__ = 'lywade'
  
  input_list = map(int,raw_input().split())
  num = input_list[0]
  num_list = input_list[1:]
  A1_list, A2_list, A3_list, A4_list, A5_list= [],[],[],[],[]
  
  for i in num_list:
      if i % 5 == 0:
          A1_list.append(i)
      elif i % 5 == 1:
          A2_list.append(i)
      elif i % 5 == 2:
          A3_list.append(i)
      elif i % 5 == 3:
          A4_list.append(i)
      else:
          A5_list.append(i)
  
  def A1(list):
      count = 0
      for i in list:
          if i % 2 == 0:
              count += i
      if count == 0:
          return 'N'
      return count
  
  def A2(list):
      if not list:
          return 'N'
      num = len(list)
      count = 0
      for i in range(num):
          count += (-1)**i *list[i]
      return count
  
  def A3(list):
      if not list:
          return 'N'
      return len(list)
  
  def A4(list):
      if not list:
          return 'N'
      result = float(sum(list))/len(list)
      return "%.1f" %result
  
  def A5(list):
      if not list:
          return 'N'
      return max(list)
  
  print A1(A1_list),A2(A2_list),A3(A3_list),A4(A4_list),A5(A5_list)