PAT1064 朋友数(20)

1064. 朋友数(20)

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

如果两个整数各位数字的和是一样的，则被称为是“朋友数”，而那个公共的和就是它们的“朋友证号”。例如123和51就是朋友数，因为1+2+3 = 5+1 = 6，而6就是它们的朋友证号。给定一些整数，要求你统计一下它们中有多少个不同的朋友证号。注意：我们默认一个整数自己是自己的朋友。

输入格式：

输入第一行给出正整数N。随后一行给出N个正整数，数字间以空格分隔。题目保证所有数字小于10⁴。

输出格式：

首先第一行输出给定数字中不同的朋友证号的个数；随后一行按递增顺序输出这些朋友证号，数字间隔一个空格，且行末不得有多余空格

输入样例：

8
123 899 51 998 27 33 36 12

输出样例：

4
3 6 9 26

---

提交代码

【思路】：

最大的位数值和为36，就设一个数组存一下，标记是否有出现过这个和就好了，本来想用set去重的，结果有一个样例超时了

但是没关系，重温了set的用法，功能，去重，自动排序

【超时代码】：

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<set>
using namespace std;
int de(int num)
{
    int sum=0,a;
    while(num!=0){
        a=num%10;
        sum+=a;
        num/=10;
    }
    return sum;
}
int main()
{
    set<int>pp;
    int a[10010],n;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    for(int i=0;i<n;i++)
    {
        for(int j=i;j<n;j++)
        {
            if(de(a[i])==de(a[j])){
                pp.insert(de(a[i]));
            }
        }
    }
    set<int>::iterator it;
    printf("%d\n",pp.size());
    int cn=1;
    for(it=pp.begin();it!=pp.end();it++)
    {
        if(cn!=pp.size())
            printf("%d ",*it);
        else printf("%d",*it);
        cn++;
    }
    return 0;
}

【AC代码】：

#include <stdio.h>
int main()
{
    int N, m, sum, friendID[37]={0},count=0;
    scanf("%d", &N);
    for(int i = 0; i < N; i++)
    {
        scanf("%d", &m);
        for(sum = 0; m; m /= 10)
            sum += m % 10;
        if(!friendID[sum]) {
            friendID[sum]=1;
            count++;
        }
    }
    printf("%d\n",count);
    for(int i=1;i<37;i++)
        if(friendID[i])
            printf("%d%c",i,--count?' ':'\0');
    return 0;
}