프로그래머스 - 테이블 해시 함수 - C++

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

프로그래머스

 

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int solution(vector<vector<int>> data, int col, int row_begin, int row_end) {
    int answer = 0;
    
    sort(data.begin(),data.end(),
         [col](const vector<int>& a,const vector<int>&b){
            if(a[col-1]!=b[col-1])
                return a[col-1]<b[col-1];
             else
                 return a[0]>b[0];
         });
    for(int i=row_begin;i<=row_end;i++)
    {
        int sum=0;
        for(int item : data[i-1])
            sum+=item%i;
        if(i==row_begin)
            answer = sum;
        else if(i!=row_begin)
            answer = sum^answer;
        
    }
    return answer;
}

주어진 data를 조건에 맞게 정렬하고 S_i 를 각각 구해서 answer XOR 연산으로 누적한 결과를 리턴하는 문제이다. 

정렬은 sort함수로 사용했다. 이 때, 람다함수로 compare함수를 넘겨줬다. 

col을 사용해야하기 때문에 [] 안에 col을 넣고 () 에 매개변수를 넣는다. 

 

그리고 for문을 돌며 합계를 구하고 answer에 xor연산을 한다.