【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;
}

另一种写法,给迭代器赋值的函数,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;
}
发布了519 篇原创文章 · 获赞 46 · 访问量 30万+

猜你喜欢

转载自blog.csdn.net/qq_33957603/article/details/104490956