[백준 - 1181번] 단어 정렬

• set 사용
• set 정렬 기준 변경 -> 구조체 써서 바꿀 수 있음
• 정의 할 때 set<자료형, 기준> 이렇게 하면됨

 

#include <iostream>
#include <string>
#include <set>
using namespace std;

struct Compare {
	bool operator() (const string& a, const string& b) const {
		// 같은 경우는 less정렬인가 봄(불확실)
		if (a.size() == b.size())
			return a < b;
		else
			return a.size() < b.size();
	}
};

int main(void)
{
	int N;
	cin >> N;
	string str;
	set<string, Compare> s;

	for (int i = 0; i < N; i++)
	{
		cin >> str;
		s.insert(str);

	}

	for (set <string>::iterator it = s.begin(); it !=s.end(); ++it)
	{
		cout << *it << '\n';
	}
	return 0;
}

 

[참고 자료]

https://chanhuiseok.github.io/posts/algo-46/