feat: [프로그래머스][greedy] 체육복 - 배열 버전 추가

Author: dev-oil

ts-programmers/greedy/001_체육복.ts

@@ -1,6 +1,51 @@
 // https://school.programmers.co.kr/learn/courses/30/lessons/42862
 
 {
+  // 세번째 풀이 배열 버전
+  function solution4(n: number, lost: number[], reserve: number[]): number {
+    let answer = 0;
+
+    const studentArr: number[][] = [];
+
+    for (let i = 0; i < n; i++) {
+      studentArr.push([i + 1, 1]);
+    }
+
+    // 잃은개수 빼기
+    for (const student of lost) {
+      const idx = studentArr.findIndex(([num]) => num === student);
+      if (idx !== -1) studentArr[idx][1] -= 1;
+    }
+
+    // 여유분 더하기
+    for (const student of reserve) {
+      const idx = studentArr.findIndex(([num]) => num === student);
+      if (idx !== -1) studentArr[idx][1] += 1;
+    }
+
+    for (let i = 0; i < studentArr.length; i++) {
+      const [, count] = studentArr[i];
+
+      if (count === 0) {
+        // 왼쪽 학생
+        if (i > 0 && studentArr[i - 1][1] === 2) {
+          studentArr[i - 1][1] -= 1;
+          studentArr[i][1] += 1;
+        }
+        // 오른쪽 학생
+        else if (i < n - 1 && studentArr[i + 1][1] === 2) {
+          studentArr[i + 1][1] -= 1;
+          studentArr[i][1] += 1;
+        }
+      }
+    }
+
+    return studentArr.filter(([, v]) => v >= 1).length;
+  }
+
+  // test
+  console.log(solution4(5, [2, 2, 2, 4], [1, 2, 3, 3, 5])); // 4
+
   // 세번째 풀이 - 여러 벌 도난 / 여러 벌 보유의 경우
   // ex) lost = [2, 2, 4] / reserve = [1, 2, 3, 3, 5]
 
@@ -19,6 +64,7 @@
     for (let i = 1; i <= n; i++) {
       studentMap.set(i, 1);
     }
+    console.log(studentMap);
 
     // 잃은개수 빼기
     for (const student of lost) {
@@ -51,7 +97,7 @@
   }
 
   // test
-  console.log(solution3(5, [2, 2, 4], [1, 2, 3, 3, 5])); // 5
+  // console.log(solution3(5, [2, 2, 4], [1, 2, 3, 3, 5])); // 5
 
   // 두번째 풀이
   function solution2(n: number, lost: number[], reserve: number[]): number {

ts-programmers/greedy/003_큰_수_만들기.ts

@@ -7,12 +7,13 @@
     for (const char of number) {
       const num = Number(char);
 
+      // 이름 붙이기
+      // 스택의 마지막 숫자가 현재 숫자보다 작으면 제거한다 (더 큰 숫자를 앞으로 배치)
+      // 제거 가능한 기회(k)가 남아있는 동안 반복한다.
       while (k > 0 && answer.length && answer.at(-1)! < num) {
         answer.pop();
         k--;
       }
-
-      answer.push(num);
     }
 
     // 만약 k 가 남았을 경우 뒤에서 잘라줘야함
@@ -25,4 +26,51 @@
 
   // test
   console.log(solution('4177252841', 4)); // 775841
+
+  // solution2 - 함수형으로 분할해보고 assert 까지 적용해보기
+  function isDescending(arr: number[]): boolean {
+    for (let i = 0; i < arr.length - 1; i++) {
+      if (arr[i] < arr[i + 1]) return false;
+    }
+    return true;
+  }
+
+  function optimizeStack(stack: number[], num: number, k: number): number {
+    // while 없애면 assert 터트릴 수 있음
+    while (k > 0 && stack.length && stack.at(-1)! < num) {
+      stack.pop();
+      k--;
+    }
+    stack.push(num);
+    console.assert(
+      isDescending(stack),
+      '스택의 숫자는 항상 내림차순이어야 함당'
+    );
+    return k;
+  }
+
+  function solution2(number: string, k: number): string {
+    const answer: number[] = [];
+
+    for (const char of number) {
+      const num = Number(char);
+      k = optimizeStack(answer, num, k);
+    }
+
+    if (k > 0) {
+      answer.splice(-k);
+    }
+
+    return answer.join('');
+  }
+
+  // test
+  console.log(solution2('4177252841', 4)); // 775841
 }
+// // ok, k > 0
+// [4]
+// [4, 1]
+
+// // not ok
+// [7, 2, 5]
+// [1,2,3]