[백준] 단어정렬 1181번 문제풀이

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

1181번: 단어 정렬

안녕하세요 주인장 H입니다. 

 

문제

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

1. 길이가 짧은 것부터

2. 길이가 같으면 사전 순으로

입력

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

 

N의 갯수가 20,000이므로 퀵정렬을 사용하여 정리하겠습니다.

sort()함수를 사용하겠습니다.

#include <iostream>
#include <algorithm>
using namespace std;
int number;
string a[20000];

//길이비교하여 우선순위를 줍니다. 
bool compare(string a, string b) {
	if (a.length() < b.length()) {
		return 1;
	}
	else if (a.length() > b.length()) {
		return 0;
	}
//c++의 경우 string은 부등호 비교시 사전순으로 알아서 비교해줍니다. 
	else {
		return a < b;
	}
}

int main(void) {
	cin >> number;
	for (int i = 0; i < number; i++) {
		cin >> a[i];
	}
	sort(a, a + number, compare);
	for (int i = 0; i < number; i++) {
		if (i > 0 && a[i] == a[i - 1]) continue;
		cout << a[i] << '\n';
	}
}