백준 1181번 : 단어 정렬 [C++]

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

1181번: 단어 정렬

소스 코드 :

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

bool compare(string a, string b)
{
	int i = 0;
	if (a.length() == b.length())
	{
		for (int i = 0; i < a.length(); i++)
		{
			if (a[i] != b[i])
				return a[i] < b[i];
		}
	}
	return a.length() < b.length();
}

int main()
{
	int num = 0;
	string input;
	vector<string> v;

	cin >> num;
	for (int i = 0; i < num; i++)
	{
		cin >> input;
		v.push_back(input);
	}

	sort(v.begin(), v.end(), compare);
	
	cout << v[0] << endl;
	for (int i = 1; i < num; i++)
	{
		if (v[i - 1] == v[i])
			continue;
		cout << v[i] << endl;
	}
}

마무리 : 이 문제는 정렬 문제로 단어의 길이 순으로 정렬하는 대신 같은 길이면 사전 순으로 정렬하는 문제이다. 그냥 sort를 하면 사전 순으로 배열하기 때문에 compare 함수를 이용해서 길이 순으로 정렬하게 만들어주며 만약 길이가 같으면 사전 순으로 정렬하게 만들었다. if문을 통해서 만약 같은 단어가 있으면 continue를 이용해서 무시하도록 만들어준다. 

 이 문제를 통해서 sort에 대한 이용법을 잘 알게 되었다.