[프로그래머스] 문자열 내 마음대로 정렬하기 C++ 풀이

https://school.programmers.co.kr/learn/courses/30/lessons/12915?language=cpp# 

프로그래머스

 

방법(1) - comp 함수를 만들기

#include <string>
#include <vector>
#include <algorithm>
using namespace std;
int idx;
bool comp(string a, string b)
{
    if(a[idx]<b[idx])
        return true;
    else if(a[idx]==b[idx])
    {
        if(a<b)
            return true;
        return false;
    }
    return false;
}

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
    idx=n;
    sort(strings.begin(),strings.end(),comp);
    answer=strings;
    return answer;
}

 

방법(2) - 사전 순 정렬 후, n번째 문자열을 기준으로 버블 정렬

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;
    sort(strings.begin(),strings.end());
    
    for(int i=0;i<strings.size();i++)
    {
        for(int j=0;j<strings.size()-i-1;j++)
        {
            if(strings[j][n]>strings[j+1][n])
            {
                string temp = strings[j+1];
                strings[j+1]=strings[j];
                strings[j]=temp;
            }
        }
    }
    
    answer=strings;
    return answer;
}