프로그래머스 - 문자열 내 마음대로 정렬하기

108번뇌·2020년 10월 2일
0

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
    vector<pair<int, string>> vector;
    for(int i=0; i<strings.size(); i++)
    {
        vector.push_back(make_pair(strings[i][n], strings[i]));
    }
    sort(vector.begin(),vector.end());
    for(int i=0; i<strings.size(); i++)
    {
        answer.push_back(vector[i].second);
    }
    return answer;
}

[방법1]
1. vector<pair<Type1, Type2>> V;
2. V.push_back(make_pair(Type1,Type2));
이후 V를 Sort함수 하면,
V.first를 가지고 Sort를 한다.
ㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡㅡ
pair 와 make_pair의 활용 잘 익혀두기!

[방법2]

#include <string>
#include <vector>
#include <algorithm>
using namespace std;
int number;
bool cmp(string a, string b)
{
    if(a[number] != b[number])
    {
        return a[number]<b[number];
    }
    else
    {
        return a<b;
    }
}

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
 number = n;
 sort(strings.begin(), strings.end(), cmp);
  for(int i=0; i<strings.size(); i++)
  {
      answer.push_back(strings[i]);
  }
    return answer;
}

[방법2]에서 Sort함수의 활용,
sort 함수 매개변수 3번째에 함수 주고,
bool cmp(Type Cmp1, Type Cmp2)
{
return Cmp1 > Cmp2; : 내림차순
}
{
return Cmp1 < Cmp2; : 오름차순
}
이 형태 자주 써먹는다. 익히기.

profile
내일 아침 눈을 떳을 때, '기대되는 오늘 하루를 만들기 위해' 나는 오늘도 생각하고 고민한다.

0개의 댓글