[프로그래머스] 단체사진 찍기

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

풀이

경우의 수를 수학으로 계산하려 했는데 숫자가 작아 직접 세어도 될 것 같다는 생각이 들었다. 그래서 완전탐색으로 모든 경우에 조건을 검사하였다.

코드

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

// 조건 확인
bool check_rule(vector<char>& friends, string rule) {
    char a = rule[0]; char b = rule[2];
    char op = rule[3]; int dist = rule[4]-'0';

    for (int i = 0; i < friends.size() - 1; i++) {
        if (friends[i] == a || friends[i] == b) {
            for (int j = i + 1; j < friends.size(); j++) {
                if (friends[j] == a || friends[j] == b) {
                    if (op == '=' && dist != j - i - 1) return false;
                    else if (op == '>' && j - i - 1 <= dist) return false;
                    else if (op == '<' && j - i - 1 >= dist) return false;
                }
            }
        }
    }
    return true;
}

int solution(int n, vector<string> data) {
    int answer = 0;
    // 카카오프렌즈를 줄세우기 위한 백터
    vector<char> friends = { 'A', 'C', 'F', 'J', 'M', 'N', 'R', 'T' };
    // 모든 경우로 줄세운다
    do {
        bool is_ok = true;
        // 데이터를 확인해서 조건에 걸리는지 검사
        for (string rule : data) {
            if (!check_rule(friends, rule)) {
                is_ok = false;
                break;
            }
        }
        // 모든 조건을 만족한 경우의 수를 센다.
        if (is_ok) answer++;
    } while (next_permutation(friends.begin(), friends.end()));
    return answer;
}