[문제]
* 본 문제는 정확성과 효율성 테스트 각각 점수가 있는 문제입니다.
카카오는 하반기 경력 개발자 공개채용을 진행 중에 있으며 현재 지원서 접수와 코딩테스트가 종료되었습니다. 이번 채용에서 지원자는 지원서 작성 시 아래와 같이 4가지 항목을 반드시 선택하도록 하였습니다.
- 코딩테스트 참여 개발언어 항목에 cpp, java, python 중 하나를 선택해야 합니다.
- 지원 직군 항목에 backend와 frontend 중 하나를 선택해야 합니다.
- 지원 경력구분 항목에 junior와 senior 중 하나를 선택해야 합니다.
- 선호하는 소울푸드로 chicken과 pizza 중 하나를 선택해야 합니다.
인재영입팀에 근무하고 있는 니니즈는 코딩테스트 결과를 분석하여 채용에 참여한 개발팀들에 제공하기 위해 지원자들의 지원 조건을 선택하면 해당 조건에 맞는 지원자가 몇 명인 지 쉽게 알 수 있는 도구를 만들고 있습니다.
예를 들어, 개발팀에서 궁금해하는 문의사항은 다음과 같은 형태가 될 수 있습니다.
코딩테스트에 java로 참여했으며, backend 직군을 선택했고, junior 경력이면서, 소울푸드로 pizza를 선택한 사람 중 코딩테스트 점수를 50점 이상 받은 지원자는 몇 명인가?
물론 이 외에도 각 개발팀의 상황에 따라 아래와 같이 다양한 형태의 문의가 있을 수 있습니다.
- 코딩테스트에 python으로 참여했으며, frontend 직군을 선택했고, senior 경력이면서, 소울푸드로 chicken을 선택한 사람 중 코딩테스트 점수를 100점 이상 받은 사람은 모두 몇 명인가?
- 코딩테스트에 cpp로 참여했으며, senior 경력이면서, 소울푸드로 pizza를 선택한 사람 중 코딩테스트 점수를 100점 이상 받은 사람은 모두 몇 명인가?
- backend 직군을 선택했고, senior 경력이면서 코딩테스트 점수를 200점 이상 받은 사람은 모두 몇 명인가?
- 소울푸드로 chicken을 선택한 사람 중 코딩테스트 점수를 250점 이상 받은 사람은 모두 몇 명인가?
- 코딩테스트 점수를 150점 이상 받은 사람은 모두 몇 명인가?
즉, 개발팀에서 궁금해하는 내용은 다음과 같은 형태를 갖습니다.
[조건]을 만족하는 사람 중 코딩테스트 점수를 X점 이상 받은 사람은 모두 몇 명인가? |
[문제]
지원자가 지원서에 입력한 4가지의 정보와 획득한 코딩테스트 점수를 하나의 문자열로 구성한 값의 배열 info, 개발팀이 궁금해하는 문의조건이 문자열 형태로 담긴 배열 query가 매개변수로 주어질 때,
각 문의조건에 해당하는 사람들의 숫자를 순서대로 배열에 담아 return 하도록 solution 함수를 완성해 주세요.
[제한사항]
- info 배열의 크기는 1 이상 50,000 이하입니다.
- info 배열 각 원소의 값은 지원자가 지원서에 입력한 4가지 값과 코딩테스트 점수를 합친 "개발언어 직군 경력 소울푸드 점수" 형식입니다.
> 개발언어는 cpp, java, python 중 하나입니다.
> 직군은 backend, frontend 중 하나입니다.
> 경력은 junior, senior 중 하나입니다.
> 소울푸드는 chicken, pizza 중 하나입니다.
> 점수는 코딩테스트 점수를 의미하며, 1 이상 100,000 이하인 자연수입니다.
> 각 단어는 공백문자(스페이스 바) 하나로 구분되어 있습니다.
- query 배열의 크기는 1 이상 100,000 이하입니다.
- query의 각 문자열은 "[조건] X" 형식입니다.
[조건]은 "개발언어 and 직군 and 경력 and 소울푸드" 형식의 문자열입니다.
> 언어는 cpp, java, python, - 중 하나입니다.
> 직군은 backend, frontend, - 중 하나입니다.
> 경력은 junior, senior, - 중 하나입니다.
> 소울푸드는 chicken, pizza, - 중 하나입니다.
> '-' 표시는 해당 조건을 고려하지 않겠다는 의미입니다.
> X는 코딩테스트 점수를 의미하며 조건을 만족하는 사람 중 X점 이상 받은 사람은 모두 몇 명인 지를 의미합니다.
> 각 단어는 공백문자(스페이스 바) 하나로 구분되어 있습니다.
> 예를 들면, "cpp and - and senior and pizza 500"은 "cpp로 코딩테스트를 봤으며, 경력은 senior 이면서 소울푸드로 pizza를 선택한 지원자 중 코딩테스트 점수를 500점 이상 받은 사람은 모두 몇 명인가?"를 의미합니다.
[입출력 예]
info | query | result |
["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"] | ["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"] | [1,1,1,1,2,4] |
입출력 예에 대한 설명
지원자 정보를 표로 나타내면 다음과 같습니다.
언어 | 직군 | 경력 | 소울 푸드 | 점수 |
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 |
"java and backend and junior and pizza 100" : java로 코딩테스트를 봤으며, backend 직군을 선택했고 junior 경력이면서 소울푸드로 pizza를 선택한 지원자 중 코딩테스트 점수를 100점 이상 받은 지원자는 1명 입니다.
"python and frontend and senior and chicken 200" : python으로 코딩테스트를 봤으며, frontend 직군을 선택했고, senior 경력이면서 소울 푸드로 chicken을 선택한 지원자 중 코딩테스트 점수를 200점 이상 받은 지원자는 1명 입니다.
"cpp and - and senior and pizza 250" : cpp로 코딩테스트를 봤으며, senior 경력이면서 소울푸드로 pizza를 선택한 지원자 중 코딩테스트 점수를 250점 이상 받은 지원자는 1명 입니다.
"- and backend and senior and - 150" : backend 직군을 선택했고, senior 경력인 지원자 중 코딩테스트 점수를 150점 이상 받은 지원자는 1명 입니다.
"- and - and - and chicken 100" : 소울푸드로 chicken을 선택한 지원자 중 코딩테스트 점수를 100점 이상을 받은 지원자는 2명 입니다.
"- and - and - and - 150" : 코딩테스트 점수를 150점 이상 받은 지원자는 4명 입니다.
>문제 풀이
1.) 입력된 info를 띄어쓰기로 쪼개서 arr배열을 만든다.
ex) "java backend junior pizza 150" info에 대해
=> arr[]={"java", "backend", "junior", "pizza", "150"};
1-1) combination(String str, int length, arr)
: 함수를 이용하여 arr로 만들 수 있는 모든 조합을 map에 넣는다.
map은 hashmap<String, ArrayList<Integer>> 구조로 Key값에는 조합으로 만들어진 문자열을 넣고, Value값에는 문자열에 따라 점수를 리스트에 추가하는 방법으로 저장한다.
//조합 (str, cond조건, arr[])
//info로 만들 수 있는 모든 조합을 만들고 str마다 점수를 리스트로 만들어서 map에 저장
public void combination(String str, int length, String[] arr){
if(length==4){
int score= Integer.parseInt(arr[4]);
if(map.containsKey(str)) map.get(str).add(score);
else{
ArrayList<Integer> list= new ArrayList<>();
list.add(score);
map.put(str, list);
}
return;
}
combination(str+"-", length+1, arr);
combination(str+arr[length], length+1, arr);
}
2.) map에 있는 점수리스트 전부 오름차순 정렬해주기
** 이부분을 놓쳐서 효율성 검사에서 시간초과가 났었다.
입력 제한사항으로 info는 5만개, query는 10만개까지 들어올 수 있다.
조건은 (cpp/java/python/ - ) x (backend/frontend/ - ) x (junior/senior/ - ) x (chicken/pizza/ - )
따라서 만들 수 있는 조합의 최대 개수라고 해봐야 4*3*3*3= 108개 밖에 안된다.
근데 query를 순회하면서 정렬을 해줄 경우 최대 10만번이나 하는 상황이 생길 수 있다.
3) query를 조건문자열과 점수로 쪼개서 원하는 [조건]에 점수리스트를 이분탐색하여 원하는 점수 이상의 지원자의 인원을 파악한다.
ex) query[0] [조건]: "javabackendjuniorpizza" / 점수: 100점
map에 key값이 "javabackendjuniorpizza"인 value 즉 점수 리스트를 이분 탐색한다.
>전체 코드
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Collections;
class Solution {
static HashMap<String, ArrayList<Integer>> map;
public int[] solution(String[] info, String[] query) {
int[] answer = new int[query.length];
map= new HashMap<>();
//1. info는 " "띄어쓰기로 쪼갤 수 있음
// 쪼개서 combination()으로 만들 수 있는 모든 조합을 map에 넣는다.
for(String str: info){
String arr[]= str.split(" ");
combination("", 0, arr);
}
//2. map에 있는 점수 리스트: 오름차순 정렬
for(String str: map.keySet()){
Collections.sort(map.get(str));
}
//3. query를 and로 쪼개서 [조건 문자열]과 점수로 나누고
// [조건]에 맞는 지원자 중에 점수 리스트를 이분탐색하여 답을 구한다.
int ansIdx=0;
for(String str: query){
String new_str= str.replace(" and ", "");
String[] q_arr= new_str.split(" ");
answer[ansIdx++]= binarySearch(q_arr[0], Integer.parseInt(q_arr[1]));
}
return answer;
}
//조합 (str, cond조건, arr[])
//info로 만들 수 있는 모든 조합을 만들고 str마다 점수를 리스트로 만들어서 map에 저장
public void combination(String str, int length, String[] arr){
if(length==4){
int score= Integer.parseInt(arr[4]);
if(map.containsKey(str)) map.get(str).add(score);
else{
ArrayList<Integer> list= new ArrayList<>();
list.add(score);
map.put(str, list);
}
return;
}
combination(str+"-", length+1, arr);
combination(str+arr[length], length+1, arr);
}
//이분탐색: 원하는 점수 이상 받은 지원자를 찾기위해
//같은 점수일 경우 가장 왼쪽에 있는 index 채택
public int binarySearch(String query, int score){
if(!map.containsKey(query)) return 0;
ArrayList<Integer> list= map.get(query);
int start=0, end=list.size()-1, mid=0;
while(start<=end){
mid= (start+end)/2;
if(list.get(mid)<score){
start= mid+1;
}else{ //score<=mid
end= mid-1;
}
}
return list.size()-start;
}//bn_Search
}
https://programmers.co.kr/learn/courses/30/lessons/72412
'알고리즘 > 프로그래머스' 카테고리의 다른 글
[프로그래머스]level 1: 최소직사각형 _Java(위클리 8주차) (0) | 2021.10.27 |
---|---|
[프로그래머스]level 3: 광고삽입_Java (0) | 2021.10.27 |
[프로그래머스]level 1: 로또 최고 순위와 최저 순위_Java (0) | 2021.10.27 |
[프로그래머스]level 1: 직업군 추천하기_Java(위클리 4주차) (0) | 2021.10.26 |
[프로그래머스] level 2: 입실퇴실_Java(위클리 7주차) (0) | 2021.10.26 |