1181 번 : 단어 정렬

문제는 다음과 같습니다.

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

1181번: 단어 정렬

 

풀이 코드는 다음과 같습니다.

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
using namespace std;

bool cmp(string s1,string s2)
{
	if(s1.length()==s2.length())
		return s1<s2;
	else
		return s1.length()<s2.length();
}
int main()
{
	int n;
	cin>>n;
	vector<string>v;
	for(int i{0};i<n;i++)
	{
		string s;
		cin>>s;
		v.push_back(s);	
	}
	sort(v.begin(),v.end(),cmp);
	cout<<v[0]<<"\n";
	for(int i{1};i<n;i++)
	{
		if(v[i-1]==v[i])
			continue;
		else
			cout<<v[i]<<"\n";
	}
	return 0;
}

cmp라는 사용자 정의 함수를 통해 정렬을 하도록 구현했습니다.

중복 제거를 해주어야 하는데 처음에는 unique를 통해 제거할까 했지만 정렬을 한 다음 제거를 해주어야 하기 때문에 차라리 정렬 이후 중복되는 것을 continue 하는 방식으로 제거해 주었습니다.