【PAT乙级】1064 朋友数 (20分 C++ 会错题意的一道题）

思路

字符串来处理，得出朋友数比较方便，用int的话还要求余很麻烦，得到朋友数之后用set容器来存，自动排序+去重，代码量直接快没了。

第一遍代码

一开始我以为朋友数必须是出现两次以上的数字才行，结果样例都没过，仔细看了一下发现原来只要是出现过的数的朋友数就行了。。题目描述感觉不是很清楚0.0

//这是一开始的，还遍历比较了一遍他们的朋友数
#include<bits/stdc++.h>
using namespace std;
int main() {
	int n;
	cin >> n;
	string s;
	int a[n] = {0};
	for(int i = 0; i < n; i++) {
		cin >> s;
		for(int j = 0; j < s.size(); j++){
			a[i] += s[j] - '0';
		}
	}
	int cnt = 0;
	set<int > t;
	for(int i = 0; i < n - 1; i++){
		for(int j = 1; j < n; j++){
			if(a[i] ==  a[j]){
				t.insert(a[i]);
			}
		}
	}
	cout << t.size() << endl;
	for(auto it = t.begin(); it != t.end(); it++){
		if(it != t.begin()) cout << " ";
		cout << *it;
	}
	return 0;
}

修改后

#include<bits/stdc++.h>
using namespace std;
int main() {
	int n, sum;
	cin >> n;
	string s;
	set<int > t;
	for(int i = 0; i < n; i++) {
		sum = 0;
		cin >> s;
		for(int j = 0; j < s.size(); j++) sum += s[j] - '0';
		t.insert(sum);
	}
	cout << t.size() << endl;
	for(auto it = t.begin(); it != t.end(); it++){
		if(it != t.begin()) cout << " ";
		cout << *it;
	}
	return 0;
}

才刷了没几道PAT乙级的题目，甲级甚至一题都没开始，等乙级做完再看吧，主要也挺忙的，突然想记录一下最近做题的过程，但写一篇博客好麻烦啊，而且自己也比较菜- -，，之前有几题还是非常有意思的，挺想记录下来，可能之后会补吧