[JAVA] 단체사진 찍기(2017 카카오코드 본선) - 프로그래머스

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

코딩테스트 연습 - 단체사진 찍기

 

-- 문제 풀이

1. 순열 구하기

2. 조건 값 계산

 

class Solution {
    static int answer = 0;
    static char[] people = {'A', 'C', 'F', 'J', 'M', 'N', 'R', 'T'};
    
    public int solution(int n, String[] data) {
        answer = 0;
        char[] ch = new char[8];
        boolean[] visit = new boolean[8];
        dfs(0, ch, visit, n, data);
        return answer;
    }
    
    void dfs(int k, char[] ch, boolean[] visit, int n, String[] data){
        if(k==8){
        	for(int i=0; i<n; i++) {
        		int interval = 0;
        		int cnt = 0;
                for(int j=0; j<8; j++){
                    if(ch[j] == data[i].charAt(0) || ch[j] == data[i].charAt(2)){
                        cnt++;
                        continue;
                    }
                    if(cnt==1){
                        interval++;
                    }else if(cnt==2){
                        break;
                    }
                }
        		if(data[i].charAt(3)=='='){
                    if(interval != data[i].charAt(4) - '0') return;
                }else if(data[i].charAt(3)=='<'){
                    if(interval >= data[i].charAt(4) - '0') return;
                }else if(data[i].charAt(3)=='>')
                    if(interval <= data[i].charAt(4) - '0') return;
        	}
        	answer++;
        	return;
        }
        
        for(int i=0; i<8; i++){
            if(!visit[i]){
                ch[k] = people[i];
                visit[i] = true;
                dfs(k+1,ch,visit,n,data);
                visit[i] = false;
            }
        }
    }
}