【PAT乙】1064 朋友数 (20分) set用法

1064 朋友数 (20分)
如果两个整数各位数字的和是一样的，则被称为是“朋友数”，而那个公共的和就是它们的“朋友证号”。例如 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

• 一共用时9mins
• 不算格式大约4mins就写完了，第一遍不能有空格set迭代器不怎么会改，第二遍要输出数量忘记了答案错误。最后A了

#include<iostream>
#include<set>
using namespace std;
int main(){
	int n; cin>>n;
	set<int>ans;
	for(int i = 1; i <= n; i++){
		int x;  cin>>x;
		int t = 0;
		while(x>0){
			t += x%10;
			x /= 10;
		}
		ans.insert(t);
	}
	cout<<ans.size()<<'\n';
	set<int>::iterator it = ans.begin(), it2 = --ans.end();
	for(; it != it2; it++){
		cout<<*it<<" ";
	}
	cout<<*it;
	return 0;
}
  
 

  
 
 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
  
21
  
22
  
23
  
24
 

另一种写法，给迭代器赋值的函数，advance指定第几个元素

#include<iostream>
#include<set>
using namespace std;
int main(){
	int n; cin>>n;
	set<int>ans;
	for(int i = 1; i <= n; i++){
		int x;  cin>>x;
		int t = 0;
		while(x>0){
			t += x%10;
			x /= 10;
		}
		ans.insert(t);
	}
	cout<<ans.size()<<'\n';
	set<int>::iterator it = ans.begin(), it2 = ans.begin();
	advance(it2,ans.size()-1);
	for(; it != it2; it++){
		cout<<*it<<" ";
	}
	cout<<*it;
	return 0;
}

  
 

  
 
 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
  
21
  
22
  
23
  
24
 

文章来源: gwj1314.blog.csdn.net，作者：小哈里，版权归原作者所有，如需转载，请联系作者。

原文链接：gwj1314.blog.csdn.net/article/details/104490956