백준 / 정렬 / 1181번 / 단어 정렬 / C++

문제

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

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

입력

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진

단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

출력

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

예제 입력 1 

13

but

i

wont

hesitate

no

more

no

more

it

cannot

wait

im

yours

예제 출력 1

i

im

it

no

but

more

wait

wont

yours

cannot

hesitate

 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

bool check(string a, string b) {
	int a_len = a.length();
	int b_len = b.length();
	
	if(a_len == b_len) return a < b;
	else return a_len < b_len;
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	
	vector<string> v;
	int n;
	cin >> n;
	
	while(n--) {
		string str;
		cin >> str;
		
		v.push_back(str);
	}
	
	sort(v.begin(), v.end(), check);
	v.erase(unique(v.begin(), v.end()), v.end());
	
	for(string str : v) {
		cout << str << '\n';
	}
	
	return 0;
}

---

원래 처음에는 pair로 first에는 길이를 넣고, second에는 문자열을 넣고 하려는데 정렬은 잘되었지만 중복삭제가

쉽지가 않아서 pair를 쓰지않았다...

 

일단 check함수에 문제에 나와있는 조건을 bool형식으로 return해준다.

그리고 이 return이 sort 함수에 매개변수로 붙게되는데, true일경우 sort를 수행한다.

한마디로 사용자가 지정한 조건에 맞춰(check함수안에 지정한 조건의 결과에 맞춰) sort가 수행되는것이다.

 

그리고 erase와 unique함수가 사용되는데, unique함수는 <algorithm>헤더에 존재한다.

unique함수를 사용하게 되면 중복되는 값들이 벡터에 맨마지막에 쓰레기값으로 붙게되고 쓰레기값의 첫번째 iterator를 반환한다.

따라서 unique를 한뒤 erase를 해주면 쓰레기값의 처음부터 벡터의 끝까지 다 지워주는것이다.(결국 중복값만 지워진다)

 

한가지 궁금증은 erase를 한후 sort를 수행하면 원하는 결과가 나오지 않았는데, 순서를 바꿔서 위 코드처럼 했을시

원하는 결과값이 나왔다.

 

조금더 찾아보고 공부해서 알아봐야 할 것 같다.