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

本人AC代码：(有待改进)

// 分类# 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 statusscanf("%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 ");}elseprintf("%d ",A1);if (flag_2 == 0) {printf("N ");}elseprintf("%d ",A2);if (flag_3 == 0) {printf("N ");}elseprintf("%d ",A3);if (flag_4 == 0) {printf("N ");}elseprintf("%0.1f ",A4);if (flag_5 == 0) {printf("N");}elseprintf("%d",A5);return 0;} 　　

改进后代码：

RRR