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

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

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

class Solution {
    public int solution(int n, String[] data) {
        GroupPicture picture = new GroupPicture(data);
        return picture.getAnswer();
    }

    public static class GroupPicture {
        private final String[] relations;
        private static final String[] people = {"A", "C", "F", "J", "M", "N", "R", "T"};
        private int answer;

        public GroupPicture(String[] data) {
            relations = data;
        }

        public int getAnswer() {
            String[] line = new String[people.length];
            boolean[] visited = new boolean[people.length];
            rotatePeople(line,visited,0);
            return answer;
        }


        private void rotatePeople(String[] line, boolean[] visited, int cur) {
            if (cur == people.length) {
                checkCondition(line);
            }else{
                for(int i = 0; i< people.length; i++) {
                    if (!visited[i]) {
                        line[cur] = people[i];
                        visited[i] = true;
                        cur+=1;
                        rotatePeople(line, visited, cur);
                        visited[i] = false;
                        cur--;
                    }
                }
            }
        }

        private boolean checkCondition(String[] line) {
            boolean flag = true;
            for(String relation : relations) {
                String left = relation.substring(0,1);
                String right = relation.substring(2,3);
                String compare = relation.substring(3,4);
                int between = Integer.parseInt(relation.substring(4,5));
                int leftIndex = find(line,left);
                int rightIndex = find(line,right);
                if (!isRightCondition(leftIndex,rightIndex, compare, between)) {
                    flag = false;
                    break;
                }
            }
            if (flag) 
                answer++;
            return flag;
        }
        

        private int find(String[] line, String element) {
            int i = 0;
            for(String s : line) {
                if (s.equals(element)) {
                    return i;
                }
                i++;
            }
            return -1;
        }

        private boolean isRightCondition(int left,int right,String compare, int between) {
            if (compare.equals("=")){
                return Math.abs(right-left)-1 == between;
            }else if(compare.equals(">")) {
                return Math.abs(right-left)-1>between;
            }else{
                return Math.abs(right-left)-1<between;
            }
        }
    }
}

GroupPicture

단체사진에 관하여 주어진 순서대로 되어있는 라인의 갯수에 대해 관리하는 클래스

 

rotatePeople

순열(permutation)을 만드는 함수로서 정해진 갯수가 되면 checkCondition의 함수를 실행한다.

 

checkCondition

주어진 조건에 맞는 라인인지 확인하는 메소드

각 글자들을 분리하여 알맞은 자료형태로 변환한 다음 isRightCondition에서 확인하는 절차를 가지고 그게 다 통과가 되면 answer를 증가시킨다.

 

find

글자의 index를 가져온다.

 

isRightCondition

해당 조건에 대하여 부합한지 확인하는 메소드이다.

 

getAnswer

요구하는 반환값으로 변환해주는 메소드이다. :D