언어/c, c++
백준(BAEKJOOB) 1181번 단어 정렬 문제 c++언어
깡 딱
2023. 1. 14. 20:16
728x90
문제
알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
- 길이가 짧은 것부터
- 길이가 같으면 사전 순으로
입력
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
출력
조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.
먼저 문제에서 조건 을 다알려줬다.
1.길이가 짧은 것부터
2. 길이가 같으면 사전 순으로(a부터..)
전역 변수로 들어갈 단어 들의 배열과 케이스를 만들어 준다.
string a[20000];
int n;
c++에서 length는
c언에서의 strlen 같은 문자의 길이를 나타 내는 것이다.
결국 a랑 b중에
a의 문자 길이가 더 짧을 경우에 우선순위가 된다 라는 뜻입니다.
bool compare(string a, string b) {
if (a.length() < b.length()) {
return 1;
}
else if (a.length() > b.length()) {
return 0;
}
else {
return a < b;
}
}
그리고 메인으로 들어와서 아까 지역변수로 선언해준 n과 배열에 입력받고
int main(){
cin >> n;
for (int i = 0; i < n; i++) {
cin >> a[i];
}
}
sort 함수로 정렬과 동시에 compare 에 적은 조건대로 짧은 순서대로 우선순위가 되어서 정렬된다.
sort(a, a + n, compare);
그리고 출력 조건에
" 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다."
가 있으니까 조건문 추가
for (int i = 0; i < n; i++) {
if (a[i] == a[i - 1]) {
continue;
}
cout << a[i] << "\n";
}
출력하기전에 만약 같은단어가 있다면 a[i]을 한번 무시하고 반복문으로 돌아간다는 뜻 입니다.
총 코드
#include <algorithm> //아 라이브러리를 쓰면 sort();
#include <iostream> //
using namespace std;
string a[20000];
int n;
bool compare(string a, string b) {
if (a.length() < b.length()) {
return 1;
}
else if (a.length() > b.length()) {
return 0;
}
else {
return a < b;
}
}
int main() {
cin >> n;
for (int i = 0; i < n; i++) {
cin >> a[i];
}
sort(a, a + n, compare);
for (int i = 0; i < n; i++) {
if (i>0 &&a[i] == a[i - 1]) {
continue;
}
cout << a[i] << "\n";
}
}
728x90