[C++] 벡터 구조체 정렬. 오름차순, 내림차순, 같을 경우 다른 요소 기준으로 순서 지정

* #include <algorithm> 으로 선언해주고, sort 함수를 사용한다.

* sort를 위해 구조체안에 반드시 operator<를 작성해줘야한다.

#include <iostream>
#include <vector>
#include <algorithm>

using std::vector;

int curSortMode = 0; // 0: 오름차순, 1: 내림차순
struct Group{
	double value1;
	int value2;
	Group(double a, int b) {
		value1 = a;
		value2 = b;
	}

	bool operator<(const Group &d) const {
	    if(curSortMode ==0){
        	if(value1 == d.value1)
            	return value2 < d.value2;   //value1이 같으면 value2를 오름차순 정렬
            
		    return value1 < d.value1;  //value1을 기준으로 오름차순정렬
	    }else{
        	if(value1 == d.value1)
            	return value2 > d.value2;   //value1이 같으면 value2를 내림차순 정렬
                
		    return value1 > d.value1;  //value1을 기준으로 내림차순 정렬렬
		}
    }

};

int main(){
    vector<Group> V;
    
    int value1 = 10;
    int value2 = 100;
    V.push_back(Group(value, a));
    
    value = 20;
    value2 = 200;
    V.push_back(Group(value2, b));
    
    value = 30;
    value2 = 300;
    V.push_back(Group(value3, c));
    
    value = 40;
    value2 = 400;
    V.push_back(Group(value4, d));
    
    curSortMode = 0; //오름차순 정렬을 위해 설정
    sort(V.begin(), V.end());
    
    for(Group v: V){
        std::cout<<"-----"<<std::endl;
        std::cout<<v.value1 << v.value2<<std::endl;
    }
    
    curSortMode = 1; //내림차순 정렬을 위해 설정
    sort(V.begin(), V.end());
    
    for(Group v: V){
        std::cout<<"-----"<<std::endl;
        std::cout<<v.value1 << v.value2<<std::endl;
    }
    
}

 

참고 링크

 

C++ 개념 더보기

Visual Studio C++ 사전