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

https://school.programmers.co.kr/learn/courses/30/lessons/12915

프로그래머스

문제에서 제시한 기준으로 정렬을 수행하면 되는데, 파라미터를 이용해야 하므로 캡쳐 블록에 & 를 넣어주어

Solution 함수 스코프 내의 변수를 참조하여 람다 함수에서 사용할 수 있도록 하였다.

 

#include <bits/stdc++.h>
#define fastio cin.tie(0)->sync_with_stdio(0)
#define ll long long
using namespace std;

vector<string> solution(vector<string> strings, int n) {
    vector<string> answer;

    answer = strings;
    
    sort(answer.begin(), answer.end(), [&](string a, string b)
        {
            if (a[n] == b[n])
            {
                return a < b;
            }
            return a[n] < b[n];
        });

    return answer;
}