분류 - 그리디 알고리즘
레벨 - 골드1
문제
기숙사에서 살고 있는 준규는 한 개의 멀티탭을 이용하고 있다. 준규는 키보드, 헤어드라이기, 핸드폰 충전기, 디지털 카메라 충전기 등 여러 개의 전기용품을 사용하면서 어쩔 수 없이 각종 전기용품의 플러그를 뺐다 꽂았다 하는 불편함을 겪고 있다. 그래서 준규는 자신의 생활 패턴을 분석하여, 자기가 사용하고 있는 전기용품의 사용순서를 알아내었고, 이를 기반으로 플러그를 빼는 횟수를 최소화하는 방법을 고안하여 보다 쾌적한 생활환경을 만들려고 한다.
예를 들어 3 구(구멍이 세 개 달린) 멀티탭을 쓸 때, 전기용품의 사용 순서가 아래와 같이 주어진다면,
- 키보드
- 헤어드라이기
- 핸드폰 충전기
- 디지털 카메라 충전기
- 키보드
- 헤어드라이기
키보드, 헤어드라이기, 핸드폰 충전기의 플러그를 순서대로 멀티탭에 꽂은 다음 디지털 카메라 충전기 플러그를 꽂기 전에 핸드폰충전기 플러그를 빼는 것이 최적일 것이므로 플러그는 한 번만 빼면 된다.
입력
첫 줄에는 멀티탭 구멍의 개수 N (1 ≤ N ≤ 100)과 전기 용품의 총 사용횟수 K (1 ≤ K ≤ 100)가 정수로 주어진다. 두 번째 줄에는 전기용품의 이름이 K 이하의 자연수로 사용 순서대로 주어진다. 각 줄의 모든 정수 사이는 공백문자로 구분되어 있다.
출력
하나씩 플러그를 빼는 최소의 횟수를 출력하시오.
예제 입력 1 복사
2 7 2 3 2 3 1 2 7
예제 출력 1 복사
2
풀면서...
이것도 C++로 이미 풀었던 문제였다. 하지만 자바스크립트로 풀려니 변수가 모두 const, let으로 돼어있어서 그런지 눈에 정말 안들어온다..ㅠㅠ 주석으로 자료형을 써놔야하나...
무튼 이래저래 풀었고 전과는 조금 다른 방법으로 풀었다.
풀이
1. N>=K 일경우 뽑을 필요가 없어서 0을 반환한다.
2. nextArr과 nextLoad는 각각 현재 인덱스의 값이 다음에 언제 나오는지를 저장하는 배열과 현재 기기숫자의 다음 인덱스를 저장하는 배열이다. 둘다 마지막 인덱스보다 1개 큰 K+1로 초기화 하였다.
2-1. arr을 거꾸로 순회하며 nextArr[i]에 nextLoad[arr[i]]를 넣어준다.
2-2. nextLoad[arr[i]]에는 arr[i](기기숫자,※인덱스아님주의)의 다음 인덱스를 저장하는데 초기에는 K+1이었다가 다음번 arr[i]를 조회 할 때는 현재 인덱스를 확인하도록 nextLoad[arr[i]]=i로 바꿔준다.
2-3. 모든 반복문을 돌면 nextArr은 기기 사용 순서 배열인 arr의 인덱스에 따라 다음번에 사용할 순서를 기록한 배열이 된다. (nextLoad는 수명을 다하셨...)
3. arr을 앞에서부터 보면서 다음과 같은 조건으로 반복한다.
3-1. 플러그에 이미 꼳혀있는가?
3-2. 플러그가 아직 다 채워지지 않았는 가?
3-3. 사용중인 플러그 중 가장 마지막에 재사용 되는 플러그는 무엇인가?
4. 3-3을 확인 하기 위해서는 nextArr[index]를 확인하면 현재 해당기기가 다음번에는 언제 사용되는지 볼 수 있다.
5. target이 0보다 크다는 것은 어느 한곳의 플러그를 빼야한다는 것이므로 답으로 출력할 cnt를 증가시킨다. (사실 return처리를 다 해줘서 조건이 필요없...)
6. 마지막으로 답을 출력한다.
코드
const input = require("fs").readFileSync("./dev/stdin").toString().trim().split("\n");
const line1 = input[0].trim().split(' ');
const N = parseInt(line1[0]), K = parseInt(line1[1]);
if(N>=K){
console.log(0);return;
}
const arr = input[1].trim().split(' ').map(function(a, i){return parseInt(a)});
let plug = Array.from({length:N}, ()=> K+1); // 인덱스를 저장
let nextArr = Array.from({length:K+1}, ()=>(K+1));
let nextLoad = Array.from({length:K+1}, ()=>(K+1));
for(let i = K-1 ; i>=0; i-- ){
nextArr[i] =nextLoad[arr[i]];
nextLoad[arr[i]] = i;
}
let cnt = 0;
arr.forEach(function(obj, ind){
let target = -1;
let amt = 0;
for(let i =0; i<N; i++){
let num = arr[plug[i]]; // 전자기기 번호
let index = plug[i]; // arr에서의 인덱스
if(num==obj){ //플러그에 꼳혀있으면
plug[i] = ind;
return;
}
if(plug[i]===K+1){
plug[i] = ind;
return;
}else if(amt<nextArr[index]){ // 다음 인덱스
amt = nextArr[index];
target = i;
}
}
if(target>=0){
cnt++;
}
plug[target]=ind;
});
console.log(cnt);
-내일 시험을 봐서 앞으로 자바스크립트로 코딩테스트를 준비할 지는 모르겠네요.. 가끔 심심하면 계속 해보는 것도 재밌긴 할거 같은데 자바로만 보는 곳도 있어서 JS보다는 JAVA를....
그럼 우테캠 1차 꼭 붙겠다는 다짐을 하며 돌아오도록 하겠습니다..
'알고리즘 > BOJ 풀이(JS)' 카테고리의 다른 글
[BOJ] 최소비용 구하기(1916) (0) | 2021.05.12 |
---|---|
[BOJ]부분합(1806) (0) | 2021.05.12 |
[BOJ] 빗물 (14719) (0) | 2021.05.07 |
[BOJ] 괄호의 값(2504) (0) | 2021.05.07 |
[BOJ]연산자 끼워넣기(14888) (1) | 2021.05.07 |
댓글