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

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

• #include<iostream>
  #include<string.h>
  #include<math.h>
  #include<iomanip>
  using namespace std;
  int main()
  {
  	int n,a;
  	int b[5];
  	memset(b,0,sizeof(b));	//将数组中的每一个数字都赋为0，头文件是<string.h> 
  	cin>>n;
  	int t=2,m=0,k=0;
  	for(int i=0;i<n;i++)
  	{
  		cin>>a;
  		if(a%5==0)
  		{
  			if(a%2==0)b[0]+=a;
  		}
  		else if(a%5==1)
  		{
  			b[1]+=pow(-1,t)*a; //pow指数函数，控制交错求和 
  			++t;
  			k=1;			//最后一个节点错误就是因为没有考虑交错求和，和可能为0的情况 
  		}
  		else if(a%5==2)
  		{
  			++b[2];
  		}
  		else if(a%5==3)
  		{
  			b[3]+=a;		//先把满足条件的元素和求出，将个数记下，最后输出的时候在求平均值 
  			++m;
  		}
  		else if(a%5==4)
  		{
  			if(a>b[4])b[4]=a;
  		}
  	}
  	for(int j=0;j<5;j++)
  	{
  		if((j==1)&&k==1) cout<<b[j];	//特别要注意交错求和可能为0的情况 
  		else if((b[j]==0)) cout<<"N";
  		else if(j==3) cout<<fixed<<setprecision(1)<<(double(b[3]))/m; //控制小数后一位 
  		else cout<<b[j];
  		if(j<4)cout<<" ";
  	}
  	return 0;
  }