[프로그래머스] 순위 검색

문제

programmers.co.kr/learn/courses/30/lessons/72412

코딩테스트 연습 - 순위 검색

문제해결방법

이 문제를 풀기 위해 [개발언어][직군][경력][소울푸드] 사이즈의 리스트 배열을 생성했다.

각 지원자들의 점수를 알맞은 배열에 추가해주고 다 추가한 뒤에 리스트들을 정렬해줬다.

문의들을 처리할 때는 '-' 인 경우는 해당 항목 전체에 대해서 계산을 해줬고 각 리스트를 이진 탐색해서 개수를 셌다.

코드

package Programmers;
 
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
 
public class Solution_순위검색 {
 
    public static List<Integer>[][][][] list = new List[3][2][2][2];
 
    public static void init() {
        for(int i=0; i<3; i++) {
            for(int j=0; j<2; j++) {
                for(int k=0; k<2; k++) {
                    for(int z=0; z<2; z++) {
                        list[i][j][k][z] = new ArrayList<Integer>();
                    }
                }
            }
        }
    }
    
    public static void sort() {
        for(int i=0; i<3; i++) {
            for(int j=0; j<2; j++) {
                for(int k=0; k<2; k++) {
                    for(int z=0; z<2; z++) {
                        Collections.sort(list[i][j][k][z]);
                    }
                }
            }
        }
    }
 
    public static int stringToInt(String s) {
        int num = 0;
        if(s.equals("cpp") || s.equals("backend") || s.equals("junior") || s.equals("chicken")) num = 0;
        else if(s.equals("java") || s.equals("frontend") || s.equals("senior") || s.equals("pizza")) num = 1;
        else num = 2;
        return num;
    }
 
    public static int getOverScore(List<Integer> list, int score) {
        int s = 0, e = list.size()-1;
        int mid;
        while(s<=e) {
            mid = (s+e)>>1;
            if(list.get(mid)<score) s = mid+1;
            else e = mid-1;
        }
        return list.size() - s;
    }
 
    public static int[] solution(String[] info, String[] query) {
        List<Integer> countList = new ArrayList<>();
        init();
        String language, field, rank, food;
        for(int i=0; i<info.length; i++) {
            String[] s = info[i].split(" ");
            language = s[0];
            field = s[1];
            rank = s[2];
            food = s[3];
            int score = Integer.parseInt(s[4]);
            list[stringToInt(language)][stringToInt(field)][stringToInt(rank)][stringToInt(food)].add(score);
        }
        sort();
        int[] la = {0, 1, 2}, fi= {0, 1}, ra= {0, 1}, fo= {0, 1};
        int lalength, filength, ralength, folength;
        for(int i=0; i<query.length; i++) {
            String[] s = query[i].split(" ");
            language = s[0];
            field = s[2];
            rank = s[4];
            food = s[6];
            int score = Integer.parseInt(s[7]);
            if(language.equals("-")) {
                la[0] = 0;
                lalength = 3;
            } else {
                la[0] = stringToInt(language);
                lalength = 1;
            }
            if(field.equals("-")) {
                fi[0] = 0;
                filength = 2;
            } else {
                fi[0] = stringToInt(field);
                filength = 1;
            }
            if(rank.equals("-")) {
                ra[0] = 0;
                ralength = 2;
            } else {
                ra[0] = stringToInt(rank);
                ralength = 1;
            }
            if(food.equals("-")) {
                fo[0] = 0;
                folength = 2;
            } else {
                fo[0] = stringToInt(food);
                folength = 1;
            }
            int sum = 0;
            for(int l=0; l<lalength; l++) {
                for(int j=0; j<filength; j++) {
                    for(int k=0; k<ralength; k++) {
                        for(int z=0; z<folength; z++) {
                            sum += getOverScore(list[la[l]][fi[j]][ra[k]][fo[z]], score);
                        }
                    }
                }
            }
            countList.add(sum);
        }
        int[] answer = new int[countList.size()];
        for(int i=0; i<countList.size(); i++) answer[i] = countList.get(i);
        return answer;
    }
    
    public static void main(String[] args) {
        String[] info = {"java backend junior pizza 150","python frontend senior chicken 210","python frontend senior chicken 150","cpp backend senior pizza 260","java backend junior chicken 80","python backend senior chicken 50"};
        String[] query = {"java and backend and junior and pizza 100","python and frontend and senior and chicken 200","cpp and - and senior and pizza 250","- and backend and senior and - 150","- and - and - and chicken 100","- and - and - and - 150"};
        System.out.println(Arrays.toString(solution(info, query)));
    }
}

코드에 대한 피드백이나 더 좋은 아이디어는 언제나 환영입니다.