PAT_B_1012 数字分类 （有待改进）

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

// 分类

# include <stdio.h>
# include <stdlib.h>

int main(void)
{
	int N;
	int A1 = 0, A2 = 0, A3 = 0, A5 = 0;
	float A4 = 0.0;
	int flag_1=0, flag_2=0, flag_3=0, flag_4=0, flag_5=0;
	int *A;
	int i;
	int temp;
	
	//  输入数据
	//***********************
	//  注：int mian 出错   [Error] ld returned 1 exit status
	scanf("%d",&N);
	A = (int *)malloc(N*(sizeof(int)));
	for (i=0; i<N; i++)
	{
		scanf("%d",&A[i]);
	}
	
	//  对数据进行分类
	for (i=0; i<N; i++)
	{
		temp = A[i]%5;
		
		switch(temp)
		{
			case 0:
				if (A[i]%10 == 0)
				{
					flag_1++;
					A1 += A[i];
				}
				break;
			case 1:
				{
					flag_2++;
					if (flag_2%2 == 1)
					{
						A2+=A[i];
					}
					else
					{
						A2-=A[i];
					}
				}
				break;
			case 2:
				flag_3++;break;
			case 3:
				{
					flag_4++;
					A4 += A[i];
				}
				break;
			case 4:
				{
					flag_5++;
					A5 = A5>A[i]?A5:A[i];
				}
				break;
			default:break;
		}
	}
	A4 = A4/flag_4;
	A3 = flag_3;
	
	if (flag_1 == 0) 
	{
		printf("N ");
	}
	else
		printf("%d ",A1);
		
	if (flag_2 == 0) 
	{
		printf("N ");
	}
	else
		printf("%d ",A2);
	
	if (flag_3 == 0) 
	{
		printf("N ");
	}
	else
		printf("%d ",A3);
		
	if (flag_4 == 0) 
	{
		printf("N ");
	}
	else
		printf("%0.1f ",A4);
	
	if (flag_5 == 0) 
	{
		printf("N");
	}
	else
		printf("%d",A5);
	
	return 0;
 }