[2017카카오코드본선] 단체사진 찍기

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

코딩테스트 연습 - 단체사진 찍기 | 프로그래머스

2017년 카카오 코드 본선 문제다.

친구들 8명의 위치에 대한 조합을 바꿔주면서 주어진 조건을 모두 만족시키는 경우의 수를 구하는 문제다.

단순히 모든 친구들을 벡터에 넣고 next_permutation으로 경우의 수를 확인해보면 되는 문제다.

떨어진 거리와 인덱스 간의 차-1이 동일한 것에 유의해야 한다.

#include <string>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;
 
// 전역 변수를 정의할 경우 함수 내에 초기화 코드를 꼭 작성해주세요.
int solution(int n, vector<string> data) {
    int answer = 0;
    vector<char> friends;
    friends.push_back('A');
    friends.push_back('C');
    friends.push_back('F');
    friends.push_back('J');
    friends.push_back('M');
    friends.push_back('N');
    friends.push_back('R');
    friends.push_back('T');
    sort(friends.begin(), friends.end());
    do{
        bool isOk = true;
        for(int i=0; i<data.size(); i++){
            char left = data[i][0];
            char right = data[i][2];
            char sign = data[i][3];
            int distance = data[i][4]-'0';
            int left_idx, right_idx;
            for(int j=0; j<8; j++){
                if(friends[j] == left){
                    left_idx = j;
                }
                if(friends[j] == right)
                    right_idx = j;
            }
            switch(sign){
                case '=':
                    if(abs(left_idx - right_idx)-1 != distance)
                        isOk = false;
                    break;
                case '<':
                    if(abs(left_idx - right_idx) - 1 >= distance)
                        isOk = false;
                    break;
                    
                case '>':
                    if(abs(left_idx - right_idx) - 1 <= distance)
                        isOk = false;
                    break;
            }       
            if(!isOk) break;
        }
        if(isOk) answer++;
    }while(next_permutation(friends.begin(), friends.end()));
    return answer;
}