백준 1181 단어정렬 [C++] - list 활용

1181번: 단어 정렬 (acmicpc.net)

1181번: 단어 정렬

---

소스코드

#include<iostream>
#include<list>

using namespace std;

bool comp(string a, string b){
  // 문자열 길이로 정렬하되, 같은 길이는 알파벳 순으로 정렬한다.
  if(a.length() != b.length())
    return (a.length() < b.length());
  else return a < b;
}

int main(){
  int num;  // 반복할 횟수
  string s; // 입력받을 문자열
  list<string> ls;  // 문자열 리스트

  cin >> num;
  for(int i = 0; i < num; i++){
    cin >> s;
    ls.push_back(s);
  }
  
  //unique - > resize
  // ***정렬 후 제거해야 떨어져있는 같은 문자열이 붙어서 제거된다.***
  ls.sort(comp);
  ls.unique();
  ls.resize(ls.size());

  list<string>::iterator iter = ls.begin(); 
  for(iter = ls.begin(); iter != ls.end(); iter++){
    cout << *iter << '\n';
  }

}

풀이과정

 

1. 입력받는 문자열을 리스트로 구성한다.

2. 길이, 알파벳 순으로 리스트를 정렬 후 중복제거한다 (인접하지 않은 요소에 대해선 unique로 중복제거되지 않는다)

 

** resize는 알고리즘에선 중요하지 않았지만 unique함수를 실행한 뒤엔 resize를 해주어야 한다.