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

 

1181번: 단어 정렬

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

www.acmicpc.net

코드 설명

cmp 함수를 선언하여 각 조건에 맞게 우선순위 큐를 선언해줍니다.

주의해야 할 사항은 우선순위 큐에서의 cmp함수는 두번째 인자가 기준입니다.

또한, 문제에서 중복값을 없애야 하기 때문에 처음에 vector를 통해 받은후 erase메서드를 이용하였습니다 ( 이를 사용할때는 sort가 우선)

 

sort(arr.begin(), arr.end());
arr.erase(unique(arr.begin(), arr.end()), arr.end());

 

😊 unique 함수는 중복함수를 뒤쪽으로 보내기 때문에 erase를 다음과 같이 활용하여 없애줍니다.

 

코드

 

#include<iostream>
#include<algorithm>
#include<queue>
#include<string>
#include<vector>
using namespace std;
int N;
vector<string> a;
struct cmp {
	bool operator()(string a, string b) { // b가 기준
		if (a.size() == b.size()) {
			return a > b;
		}
		return a.size() > b.size();

	}


};
vector<string> arr;
priority_queue<string, vector<string>, cmp> pq;
int main() {
	cin >> N;

	for (int i = 0; i < N; i++) {
		string abc;
		cin >> abc;
		arr.push_back(abc);
	}
	sort(arr.begin(), arr.end());
	arr.erase(unique(arr.begin(), arr.end()), arr.end());

	for (int i = 0; i < arr.size(); i++) {
		pq.push(arr[i]);
	}

	while (!pq.empty()) {
		cout << pq.top() << endl;
		pq.pop();


	}


	return 0;
}

 

+ Recent posts