[백준 1181] 단어 정렬

www.acmicpc.net/problem/1181

1181번: 단어 정렬

 

map 소팅을 사용해봤다.

 

문제에서 모든 단어는 한번만 등장하게 조건을 걸어서 map을 사용하여 소팅을 진행했다.

 

맵을 소팅하는 방법은 맵을 벡터에 담아서 벡터를 소팅하는 것이다.

이 때 compare함수는 map의 자료형으로 소팅하면 된다.

#include <algorithm>
#include <iostream>
#include <map>
#include <vector>

#define pp pair<string, int>

using namespace std;

map<string, int> mp;

bool compare(const pp &a, const pp &b)
{
  if (a.second == b.second)
  {
    return a.first < b.first;
  }
  return a.second < b.second;
}
int main(void)
{
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);

  int n;
  cin >> n;
  for (int i = 0; i < n; i++)
  {
    string tmp;
    cin >> tmp;
    if (mp.find(tmp) == mp.end())
    {
      mp[tmp] = tmp.length();
    }
  }

  vector<pp> v(mp.begin(), mp.end());
  sort(v.begin(), v.end(), compare);

  for (auto it : v)
  {
    cout << it.first << "\n";
  }
  return 0;
}