[BOJ] 백준_1181번_단어 정렬_ C/C++카테고리 없음2021. 11. 15. 10:51
목차
문제 출처
https://www.acmicpc.net/problem/1181
문제
알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
1. 길이가 짧은 것부터
2. 길이가 같으면 사전 순으로
입력
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
출력
조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.
시간초과 코드
//[BOJ] 1181번 단어 정렬_시간초과
#include <iostream>
#include <string>
using namespace std;
int main()
{
int n;
cin >> n;
string tmp;
string* s = new string[n];
for (int i = 0; i < n; i++)
{
cin >> s[i];
}
for (int i = 0; i < n; i++)
{
for (int j = 0; j < n-1; j++)
{
if (s[i].length() < s[j].length()) {
tmp = s[j];
s[j] = s[i];
s[i] = tmp;
}
else if (s[i].length() == s[j].length()&& s[j].compare(s[i]) > 0)
{
tmp = s[j];
s[j] = s[i];
s[i] = tmp;
}
}
}
for (int i = 0; i < n; i++)
{
if (i > 0 && s[i] == s[i - 1])
continue;
cout << s[i] << endl;
}
delete[] s;
return 0;
}
코드
//[BOJ] 1181번 단어 정렬
#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
bool compare(string a, string b) {
if (a.length() < b.length()) {
return 1;
}
else if (b.length() < a.length()) {
return 0;
}
else {
return a < b;
}
}
int main()
{
int n;
cin >> n;
string tmp;
string* s = new string[n];
for (int i = 0; i < n; i++)
{
cin >> s[i];
}
sort(s, s + n, compare);
for (int i = 0; i < n; i++)
{
if (i > 0 && s[i] == s[i - 1])
continue;
cout << s[i] << endl;
}
delete[] s;
return 0;
}
풀이 과정
n개의 원소를 가진 string 배열을 동적할당 한다. 문자열을 각 원소마다 채운 후 정렬을 한다. 이때 sort함수를 쓰지 않으면 시간초과로 처리된다. compare 조건에 따라 총 n개의 원소가 있는 배열 s를 오름차순으로 정렬해주는 sort(s, s + n, compare); 구문을 사용하여 코드를 간편하게 작성할 수 있었다.
728x90
반응형
LIST
@kdj :: Childev'note
포스팅이 좋았다면 "좋아요❤️" 또는 "구독👍🏻" 해주세요!