프로그래머스 1단계 - 문자열 내 마음대로 정렬하기 - 오답노트

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

코딩테스트 연습 - 문자열 내 마음대로 정렬하기

- 문제 파악

문자열 배열이 주어지고 정수의 인덱수 n 이 주어진다. 문자열에서 n번째 글자로 정렬하되 같은 글자일 경우 문자열 사전순으로 정렬하라

 

- 나의 접근

sort 함수를 i번째 요소와 i+1 번째 요소를 비교하여 정렬하는 로직으로 멋대로 생각하고 있었다. (실제로 어떻게 도는지는 공부해보자) 그래서 겹치는 부분만 다시 사전순으로 정렬하려고 해서 쓸데없이 코드만 길어지고 답도 맞추지 못했다.

 

- 정답

sort 에 함수를 넣어서 사용하는 방식이 익숙하지 않아서 최대한 안사용하고 버틸려고 했는데, 이젠 피하지 말고 마주해야겠다.

 

#include "bits/stdc++.h"

using namespace std;

int N;

bool cmp(string a, string b){
    if(a[N] == b[N]){
        return a < b;
    }
    else{
        return a[N] < b[N];
    }
}

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