문자열 내맘대로 정렬하기

*문자열 내맘대로 정렬하기(string) 
문자열 리스트 strings와 정수 n이 주어짐. 
문자열의 n번째 인덱스 비교. 오름차순 정렬. 
단 문자열 n번째가 같으면 전체 문자열 비교. 
 
#include <iostream>
#include <algorithm>
#include <string>
 
using namespace std; 
 
int pos; 
bool compareStr(string a, string b) 
{ 
    if(a[pos] == b[pos]) 
    { 
        return a < b; 
    } 
    return a[pos] < b[pos]; 
} 
 
vector solution(vector strings, int n) { 
    vector answer; 
 
    pos = n; 
    sort(strings.begin(), strings.end(), compareStr); 
 
    return strings; 
}