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 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;
 } 　　

改進後代碼：

RRR