학습 정리/👨‍💻 PS Study

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

무딘붓 2023. 1. 4. 23:00

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

 

1181번: 단어 정렬

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

www.acmicpc.net

[소스코드]

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
// [baekjoon] 1181번 - 단어 정렬
// 2023.01.04

bool compare(string a, string b) {
	if (a.length() == b.length()) {	// 길이가 같으면 
		return a < b;		// b가 사전순으로 a 보다 다음 순서가 되게 정렬 (= 사전 순으로 정렬)
	}
	else {					// 길이가 다르면
		return a.length() < b.length();	// b의 길이가 더 길도록 정렬 (=길이가 짧은 것이 앞으로)
	}
}

int main() {
	string str[20001];

	int n;
	cin >> n;

	for (int i = 0; i < n; i++) {
		cin >> str[i];
	}
	
	sort(str, str + n, compare);

	for (int i = 0; i < n; i++) {
		if (str[i] == str[i - 1]) {	// 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력
			continue;
		}
		cout << str[i] << "\n";
	}


	return 0;
}

c++의 sort를 이용하여 정렬하는 경우, 세번째 인자에 정렬 조건을 사용하여 정렬하는 문제입니다.

 

sort 함수에 대해서는 아래 게시글을 참고해 주세요.

https://sirius7.tistory.com/85