백준 1181번 - 단어정렬

www.acmicpc.net/problem/1181

1181번: 단어 정렬

#include <iostream>
#include <string>
#include <vector>
#include <set>
#include <algorithm>
//길이가 짧은 것부터
//길이가 같으면 사전 순으로

using namespace std;

int compare(string a, string b){
  if(a.size() == b.size()){
    return a < b;
  }
  return a.size() < b.size(); 
}

int main() {
  int n;
  cin >> n;
  cin.ignore();
  vector<string> lines(n);
  set<string> s;

  for(int i = 0; i<n; i++){
    getline(cin, lines[i]);
  }
  s.insert(lines.begin(), lines.end());

  vector<string> sorted(s.size());

  int count = 0;
  for(auto it = s.begin(); it != s.end(); it++){
		 sorted[count] = (*it);
     count++;
	}

  sort(sorted.begin(),sorted.end(),compare);

  for(int i = 0; i<sorted.size(); i++){
    cout <<sorted[i] << endl;
  }

}

 

조금 복잡하다.. 

1)먼저, lines라는 백터에 모든 값을 저장한다. 

2)s라는 셋에 넣어 중복을 제거해준다. 

3)다시 sorted라는 백터에 복사해 와서 sort라는 함수를 통해 문자열이 짧은 순으로, 문자열의 개수가 같으면 사전순으로 정렬해 주었다. 

4)출력해 준다.