백준 2870번 수학숙제 ( C++ )

문제

https://www.acmicpc.net/problem/2870

2870번: 수학숙제

 

해설

동적할당 -> vector사용

커스텀 정렬

string 크기 비교(길이가 다르면 정렬 제대로 X)

코드

  #include <bits/stdc++.h>
using namespace std;
 
int n;
string line;
string tmp;
bool check;
vector<string> ret;
 
bool cmp(string a, string b)
{
  if (a.size() == b.size())
    return b > a;
  return a.size() < b.size()
}
 
int main()
{
  cin >> n;
 
  for (int i = 0; i < n; i++)
  {
    cin >> line;
    check = false;
    for (char a : line)
    {
      tmp = a;
      if (a >= '0' && a <= '9' && check == false)
      {
        ret.push_back(tmp);
        check = true;
      }
      else if (a >= '0' && a <= '9' && check == true)
      {
        tmp = ret.back();
        if (tmp == "0")
          tmp = a;
        else
          tmp += a;
        ret.back() = tmp;
      }
      else
        check = false;
    }
  }
  sort(ret.begin(), ret.end(), cmp);
  for (auto a : ret)
  {
    cout << a << "\n";
  }
}