본문 바로가기
개인 공부/코딩테스트

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

by seowit 2020. 3. 13.

문제

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

 

1181번: 단어 정렬

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

www.acmicpc.net

소스코드

#include <algorithm>
#include <string>
#include <iostream>
#include <vector>
using namespace std;
bool compare(string x, string y) {
	if (x.length() == y.length()) {
		return x < y;
	}
	else {
		return x.length() < y.length();
	}
}

int main(void) {
	int N; cin >> N;
	vector<string> v(N);
	for (int i = 0; i < N; i++) {
		cin >> v[i];
	}
	sort(v.begin(), v.end(), compare);

	cout << v[0] << endl;
	for (int i = 1; i < N; i++) {
		if (v[i] != v[i - 1]) {
			cout << v[i] << endl;
		}
	}
}

설명

1. string을 저장하는 벡터에 입력 받은 값들을 저장한다.

2. sort로 벡터를 정렬한다. compare 함수에서 두 문자열의 길이가 같은 경우와 다른 경우를 각각 처리해준다.

3. 중복을 제거 하고 출력하기 위해 정렬 후 이전 원소와 값이 다른 값만 출력을 한다.

댓글