演算法學習之路|朋友數

kissjz發表於2018-02-28

如果兩個整數各位數字的和是一樣的,則被稱為是“朋友數”,而那個公共的和就是它們的“朋友證號”。例如123和51就是朋友數,因為1+2+3 = 5+1 = 6,而6就是它們的朋友證號。給定一些整數,要求你統計一下它們中有多少個不同的朋友證號。注意:我們預設一個整數自己是自己的朋友。

輸入格式

輸入第一行給出正整數N。隨後一行給出N個正整數,數字間以空格分隔。題目保證所有數字小於104。

輸出格式

首先第一行輸出給定數字中不同的朋友證號的個數;隨後一行按遞增順序輸出這些朋友證號,數字間隔一個空格,且行末不得有多餘空格。

輸入樣例:
8
123 899 51 998 27 33 36 12
輸出樣例:
4
3 6 9 26
解題思路

將朋友數記錄在map中的first,然後每次map[朋友數]++;

最後輸出map.size()並且依次輸出map.first成員,即依次輸出不同的朋友數即可。

#include<iostream>
#include<map>
using namespace std;
int main(){
    int n;
    cin>>n;
    int tmp;
    map<int,int>v;
    for(int i=0;i<n;i++){
        cin>>tmp;
        int sum=0;
        while(tmp!=0){
            sum+=tmp%10;
            tmp/=10;
        }
        v[sum]++;
    }
    cout<<v.size()<<endl;
    for(map<int,int>::iterator i=v.begin();;){
        cout<<i->first;
        i++;
        if(i==v.end()){
            break;
        }
        cout<<" ";
    }
}


相關文章