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;
}
'알고리즘 공부 > 백준' 카테고리의 다른 글
[백준] 마법사 상어와 비바라기(21610 /C++) (0) | 2021.06.04 |
---|---|
[백준] 연구소 (14502번 / C++) (0) | 2021.06.01 |
[백준] 14425 문자열 집합(c++) (0) | 2021.05.18 |
[백준] 1158 요세푸스 문제(C++) (0) | 2021.05.18 |
[백준] 9663 N-Queen (C++) (0) | 2021.05.17 |