PAT (Basic Level) Practice 1012 数字分类 C++

#include<iostream>
#include<iomanip>

using namespace std;

int main(){
	
	int N;
	cin>>N;
	
	int A1=0,A2[1000]={0},A3=0,A4[1000]={0},A5=0;
	int a;
	int i,j=0,k=0;
	for(i=0; i<N; i++){
		cin>>a;
		if(a%5==0 && a%2==0){
			A1+=a;
		}else if(a%5==2){
			A3++;
		}else if(a%5==4 && a>A5){
			A5=a;
		}else if(a%5==1){
			A2[j++]=a;
		}else if(a%5==3){
			A4[k++]=a;
		}
	}
	double sum=0;
	double num=k;
	double aver=0;
	for(i=0; i<k; i++){
		sum+=A4[i];
	}if(num!=0){
		aver=sum/num;	
	}
	int f=1;
	sum=0;
	for(i=0; i<j; i++){
		if(f){
			sum+=A2[i];
			f=0;
		}else{
			sum-=A2[i];
			f=1;
		}
	}
	double ans[6];
	ans[1]=A1; ans[2]=sum; ans[3]=A3; ans[4]=aver; ans[5]=A5;
	if(ans[1]==0){
		cout<<"N";
	}else{
		cout<<ans[1];
	}
	for(i=2; i<=5; i++){
		if(ans[i]==0 && !(i==2 && j!=0)){
			cout<<" N";
		}else if(i==4){
			cout<<setiosflags(ios::fixed)<<setprecision(1)<<" "<<ans[4];
		}else{
			cout<<setiosflags(ios::fixed)<<setprecision(0)<<" "<<ans[i];
		}
	}
	
	
	
	return 0;
}

最后一个（第8个）测试点卡了一会儿，容易被遗漏的情况是A2的结果可能正负刚好抵消，所以是0，这种情况下应该输出0，不应该输出N，给一个测试案例~

2 6 6

输出应该是

N 0 N N N

其他应该没什么问题，刚开始学c++，通过这题还学到了如何输出的时候控制小数的位数~