[leetCode/JS] 880. Decoded String at Index

난이도 [ 🤔 ] Medium

문제 설명

You are given an encoded string s. To decode the string to a tape, the encoded string is read one character at a time and the following steps are taken:

• If the character read is a letter, that letter is written onto the tape.
• If the character read is a digit d, the entire current tape is repeatedly written d - 1 more times in total.

Given an integer k, return the kth letter (1-indexed) in the decoded string.

 

인코딩된 문자열 s가 주어집니다. 
이 문자열을 테이프에 디코딩하기 위해 인코딩된 문자열을 한 번에 한 문자씩 읽고 다음 단계를 수행합니다:
- 읽은 문자가 문자인 경우 해당 문자가 테이프에 기록됩니다.
- 읽은 문자가 숫자 d인 경우 현재 테이프 전체가 d-1번 반복 된다.
정수 k가 주어지면 디코딩된 문자열에서 k번째 문자를 반환합니다.

 

 

 

 

입출력 예

Example 1:

Input: s = "leet2code3", k = 10
Output: "o"
Explanation: The decoded string is "leetleetcodeleetleetcodeleetleetcode".
The 10th letter in the string is "o".

 

Example 2:

Input: s = "ha22", k = 5
Output: "h"
Explanation: The decoded string is "hahahaha".
The 5th letter is "h".

 

Example 3:

Input: s = "a2345678999999999999999", k = 1
Output: "a"
Explanation: The decoded string is "a" repeated 8301530446056247680 times.
The 1st letter is "a".

 

Constraints

• 2 <= s.length <= 100
• s consists of lowercase English letters and digits 2 through 9.
• s starts with a letter.
• 1 <= k <= 109
• It is guaranteed that k is less than or equal to the length of the decoded string.
• The decoded string is guaranteed to have less than 263 letters.

---

내 솔루션

• 처음엔 로직대로 string을 계속 늘려서 return s[k-1]로 작업하려고함.
• 너무 커서 메모리 이슈로 아예 통과 조차 못함. 애초에 로직도 잘못이해서 'leet10' 이면 leet가 10번 반복 된다고 생각함. 문제에서 2~9 사이 숫자라고 알려주었지만.. 문제를 잘 읽어보자..
• 고민 중에 숫자가 d-1번 반큼 반복된다 라는 걸로 k 근처로 올때까지 반복해서 제거하는 방식을 떠올림.

var decodeAtIndex = function(s, k) {
    let size = 0;
    for (const letter of s) {
      size = isNaN(letter) ? size + 1 : letter * size;
    }
    for (let i = s.length - 1; i >= 0; i--) {
        let char = s[i];
        k %= size

        if (k === 0 && isNaN(char)) {
            return char;
        } else if (!isNaN(char)) {
            size /= +char;
        } else {
            size--;
        }
    }
    return s[1];
}

 

감상평

• 큰 틀에서는 많이 안바꿨지만 제출 할때마다 예외처리에 엄청 골치아팠다