PAT乙级 1012 数字分类

#include<iostream>
#include<vector>
#include<algorithm>
#include<iomanip>
#include<cstdio>
using namespace std;
int main()
{
	vector<int> a[5];
	int N,num,a1=0,a2=0,a5=0;
	double a4=0.0;
	cin>>N;
	while(N--)
	{
		cin>>num;
		a[num%5].push_back(num);
	}
	for(int i=0;i<5;i++)
	{
		for(int j=0;j<a[i].size();j++)
		{
			if(i==0 && a[i][j]%2==0)	a1+=a[i][j];
			if(i==1 && j%2==0)	a2+=a[i][j];
			if(i==1 && j%2==1)	a2-=a[i][j];
			if(i==3)	a4+=a[i][j];
			if(i==4 && a5<a[i][j])	a5=a[i][j];
		}
	}
	for(int i=0;i<5;i++)
	{
		if(i!=0)	cout<<" ";
		if(i==0 && a1==0 || i!=0 && a[i].size()==0)
		{
			cout<<"N";
			continue;
		}
		if(i==0) cout<<a1;
		if(i==1) cout<<a2;
		if(i==2) cout<<a[2].size();
		if(i==3) printf("%.1f",a4/a[3].size());
		if(i==4) cout<<a5;
	}
	return 0;
}