[백준 1181/c++] 단어 정렬

중복을 허락하지 않고 커스텀 조건으로 정렬 - 아하 set을 쓰면 되겠군!

set을 사용한 커스텀 정렬은 처음 써보았다 set의 템플릿에 맞게 struct 의 내부를 원하는 조건으로 커스텀해준뒤

입력을 받아 set에 넣으면 원하는 조건으로 정렬이 된다. 그대로 출력하면 완성이다!

https://www.acmicpc.net/problem/1181

1181번: 단어 정렬

#include <bits/stdc++.h>
using namespace std;

struct tok
{
	bool operator() (const string& left, const string& right) const
	{
		if(left.size() == right.size()) {
			return left < right;
		} else {
			return left.size() < right.size();
		}
	}
};


int main(){
	
	int n;
	string s;
	cin >> n;
	
	set<string, tok> st;
	
	
	for(int i=0; i<n; i++){
		cin >> s;
		st.insert(s);
	}
	
	for(auto k : st){
		cout << k << "\n";
	}
	
	return 0;
}