1. 程式人生 > >PAT乙級 朋友數

PAT乙級 朋友數

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

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

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

8
123 899 51 998 27 33 36 12

輸出樣例:

4
3 6 9 26

作者: CHEN, Yue
單位: 浙江大學
時間限制: 400 ms
記憶體限制: 64 MB
程式碼長度限制: 16 KB

#include <iostream>
#include <set>

using namespace std;

int main()
{
	int n;
	cin>>n;
	string str;
	int i,j;
	set<int> s;
	int sum[n]={0};
	for(i=0;i<n;i++)
	{
		cin>>str;
		sum[i]=0;
		for(j=0;j<str.length();j++)
		{
			sum[i]+=str[j]-'0';
		}
	}
	for(i=0;i<n;i++)
		s.insert(sum[i]);
	cout<<s.size()<<endl;
	set<int>::iterator it = s.begin();
	cout<<*it;
	it++;
	for(;it != s.end() ; it ++)
	cout<<" "<<*it;		
	getchar(); 
	return 0;
}

這道題求和沒問題,難在如何將所求的和去重並排序,剛開始我只是想著自己寫個方法實現去重,忘記了STL中的set容器(老實說,對於STL我還沒有完全熟悉,應該再看看,不能半途而廢。);本來覺得實現去重並排序就好了,輸出還有要求,要求行末不得出現多餘的空格,(姥姥是知大家都用set容器做才這樣要求的吧,(⊙o⊙)…),以前都是先按正常要求輸出前n-1個元素,再特別處理最後一個,但是對於set容器來說行不通,所以要換個想法,先特別輸出第一個元素,然後再輸出空格和剩下的元素。
呃呃,笨死了,以後要動點腦筋,大傻子。