1064. 朋友数(20)

最后一个超时

#include<set>
#include<cstdio>
using namespace std;
int fun(int x)
{
	int sum=0;
	while(x>0)
	{
		sum+=x%10;x/=10;
	}
	return sum;
}
int main()
{
	int n,a[10000+10];
	set<int> s;
	//freopen("1064.txt","r",stdin);
	scanf("%d",&n);
	for(int i=0;i<n;i++)
	{
		scanf("%d",&a[i]);
		s.insert(fun(a[i]));
	 } 
	int sum1;
	 for(int i=0;i<n;i++)
	 {
	 	sum1=fun(a[i]);
	 	for(int j=0;j<n;j++)
	 	{
	 		if(a[j]==a[i])	continue;
	 		if(sum1==fun(a[j]))
	 		{
	 			//printf("%d\n",sum1);
	 			s.insert(sum1);
			 }
		 }
	 }
	 set<int>:: iterator it;
	 printf("%d\n",s.size());
	 int flag=0;
	 for(it=s.begin();it!=s.end();it++)
	 {
	 	if(flag==0)
	 	{
	 		flag=!flag;
	 		printf("%d",*it); 
		 }	 		
	 	else 
	 		printf(" %d",*it); 
	 }

	return 0;
 }