【PAT1012B】数字分类

1012 数字分类 (20分)

给定一系列正整数，请按要求对数字进行分类，并输出以下 5 个数字：

• A​1​​ = 能被 5 整除的数字中所有偶数的和；
• A​2​​ = 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算 n​1​​−n​2​​+n​3​​−n​4​​⋯；
• A​3​​ = 被 5 除后余 2 的数字的个数；
• A​4​​ = 被 5 除后余 3 的数字的平均数，精确到小数点后 1 位；
• A​5​​ = 被 5 除后余 4 的数字中最大数字。

输入格式：

每个输入包含 1 个测试用例。每个测试用例先给出一个不超过 1000 的正整数 N，随后给出 N 个不超过 1000 的待分类的正整数。数字间以空格分隔。

输出格式：

对给定的 N 个正整数，按题目要求计算 A​1​​~A​5​​ 并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。

若其中某一类数字不存在，则在相应位置输出 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

作者: CHEN, Yue

单位: 浙江大学

时间限制: 400 ms

内存限制: 64 MB

code

#include<iostream>
#include<vector>
#include<cmath>
#include<algorithm>
#include<string>
#include<cstring>
#include<cstdlib>
using namespace std;
int main()
{
	int n;
	cin>>n;
	vector<int> a;
	int temp=0;
	for(int i=0;i<n;i++)
	{
		cin>>temp;
		a.push_back(temp);
	}
	int ans[6]={0},cnt1=0,cnt2=0;
	for(int i=0;i<n;i++)
	{
		if(a[i]%5==0)
		{
			if (a[i]%2==0)
			{
				ans[1]+=a[i];
			}
		}
		else if(a[i]%5==1)
		{
			if (cnt1%2==0)
			{
				ans[2]+=a[i];
			}
			else
				ans[2]-=a[i];
			cnt1++;
		}
		else if (a[i]%5==2)
		{
			ans[3]++;
		}
		else if (a[i]%5==3)
		{
			ans[4]+=a[i];
			cnt2++;
		}
		else
		{
			ans[5]=max(ans[5],a[i]);
		}
	}
	float temp4=0.0;
	if (cnt2!=0)
	{
		temp4=float(ans[4])/(float)cnt2;
	}
	for(int i=1;i<4;i++)
	{
		if(i==2)
		{
			if (cnt1==0)
			{
				cout<<"N ";
			}
			else
			{
				cout<<ans[i]<<" ";
			}
		}
		else
		{
			if (ans[i]==0)
			{
				cout<<"N ";
			}
			else
			{
				cout<<ans[i]<<" ";
			}
		}
		
	}
	if (ans[4]==0)
	{
		cout<<"N ";
	}
	else
	{
		printf("%.1f ",temp4);
	}
	if (ans[5]==0)
	{
		cout<<"N";
	}
	else
		cout<<ans[5];
	return 0;

}