整数的分类处理 C++

题目描述

给定 N 个正整数，要求你从中得到下列三种计算结果：

• A1 = 能被 3 整除的最大整数
• A2 = 存在整数 K 使之可以表示为 3K+1 的整数的个数
• A3 = 存在整数 K 使之可以表示为 3K+2 的所有整数的平均值（精确到小数点后 1 位）

输入

输入首先在第一行给出一个正整数 N，随后一行给出 N 个正整数。所有数字都不超过 100，同行数字以空格分隔。

输出

在一行中顺序输出 A1、A2、A3的值，其间以 1 个空格分隔。如果某个数字不存在，则对应输出NONE。

输入样例1 

8 5 8 7 6 9 1 3 10

输出样例1

9 3 6.5

思路分析

先解决主要问题：

A1 = 能被 3 整除的最大整数：先对数组排序，这里为了方便三个一起判断就从小到大排序，碰到能被 3 整除的就刷新A1的值；

A2 = 存在整数 K 使之可以表示为 3K+1 的整数的个数：其实就是找和3取余余数为1的数的个数，数就完事了；

A3 = 存在整数 K 使之可以表示为 3K+2 的所有整数的平均值（精确到小数点后 1 位）：这个和第二个类似，找到和3取余余数为2的数加起来求平均值。

输出NONE的问题：

完全可以设一个flag或者tag来判断有没有，但因为全是正整数，所以我就根据A的值来判断了，A1设计初始值为负数，如果没有找到，那么仍是负数，A2和A3是数数的，如果为0，说明没有找到。

AC代码

#include<bits/stdc++.h>
using namespace std;
int main(){
	int N,A1=-1,A2=0,A3=0,A3sum=0;
	cin>>N;
	int number[N];
	for(int i=0;i<N;i++)
	cin>>number[i];
	sort(number,number+N,less<int>());
	for(auto&it:number){
		if(it%3==0){
			A1=it;
		}else if(it%3==1){
			A2++;
		}else if(it%3==2){
			A3sum+=it;
			A3++;
		}
	}
	if(A1==-1)
	cout<<"NONE ";
	else cout<<A1<<' ';
	if(A2)
	cout<<A2<<' ';
	else cout<<"NONE ";
	if(A3)
	cout<<A3sum/(A3*1.0);
	else cout<<"NONE";
}