#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; : 오름차순
}
이 형태 자주 써먹는다. 익히기.