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

문제

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.

1. 길이가 짧은 것부터
2. 길이가 같으면 사전 순으로

 

입력

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

 

출력

조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.

 

코드

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>

using namespace std;

int N;
vector<string> v;

int cmp(string &a, string &b)
{
	if (a.size() != b.size())
		return a.size() < b.size();
	return a < b;
}

int main()
{
	int i;
	string s;

	cin >> N;
	for (i = 0; i < N; i++)
	{
		cin >> s;
		v.push_back(s);
	}
	sort(v.begin(), v.end(), cmp);
	cout << v.front() << endl;
	for (i = 1; i < v.size(); i++)
	{
		if (v[i - 1].compare(v[i]) == 0)
			continue;
		cout << v[i] << endl;
	}
	return 0;
}

 

비교 함수만 구현하면 되는 문제이다. 모든 입력값을 벡터에 저장한 다음 기준에 따라 정렬하면 된다. 출력할 때 같은 단어는 한 번만 출력해야 한다. 이미 정렬을 했기 때문에 같은 단어는 연속적으로 배치되어 있을 것이다. 따라서 이전 값을 비교해 같은 경우는 출력하지 않도록 하면 된다.

 

링크

1181번: 단어 정렬

inbdni/Baekjoon