PAT Basic 1064. 朋友数

1. 题目描述：

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

2. 输入格式：

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

3. 输出格式：

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

4. 输入样例：

8
123 899 51 998 27 33 36 12

5. 输出样例：

4
3 6 9 26

6. 性能要求：

Code Size Limit
16 KB
Time Limit
400 ms
Memory Limit
64 MB

思路：

除草题，按照题意编写即可。因为题目保证正整数最大为9999，所以朋友证号的范围为\(1 \sim 36\)，这里定义一个大小37的int型数组friendId用于记录出现的朋友证号，方便下标访问，并且第一个元素friendId[0]用于记录有多少个不同的朋友证号。

My Code:

#include <stdio.h>

int main(void)
{
    int numCount = 0;
    int friendId[36+1] = {0}; // the maximum positive number is 9999, thus the max friendId is 36, use first element to record idCount.
    int temp=0;
    int i=0; // iterator
    int sum=0;
    int firstBlood=0; // output space flag
    
    scanf("%d", &numCount);
    for(i=0; i<numCount; ++i)
    {
        scanf("%d", &temp);
        while(temp)
        {
            sum += temp%10;
            temp /= 10;
        }
        if(!friendId[sum]) // first appear
        {
            ++friendId[sum];
            ++friendId[0]; // increase idCount
        }
        sum=0;
    }
    
    printf("%d\n", friendId[0]);
    
    for(i=1; i<=36; ++i)
    {
        if(friendId[i])
        {
            if(firstBlood)
            {
                printf(" %d", i);
            }

            else
            {
                firstBlood = 1;
                printf("%d", i);
            }
        }
    }
    printf("\n");
    
    return 0;
}