[자료구조] 스택 (Stack) - 브라우저 뒤로가기 앞으로가기 with Javascript

해당 문제는 코드스테이츠의 Toy problem을 출처로 합니다.

 

문제

개발자가 되고 싶은 김코딩은 자료구조를 공부하고 있습니다. 인터넷 브라우저를 통해 스택에 대해 검색을 하면서 다양한 페이지에 접속하게 되었는데 "뒤로 가기", "앞으로 가기"를 반복하면서 여러 페이지를 참고하고 있었습니다.

그런데 새로운 페이지를 접속하게 되면 "앞으로 가기" 버튼이 비활성화돼서 다시 보고 싶던 페이지로 갈 수 없었습니다. 이러기를 반복하다가 김코딩은 스택 자료구조를 떠올리게 되었습니다.

브라우저에서 "뒤로 가기", "앞으로 가기" 기능이 어떻게 구현되는지 궁금해진 김코딩은 몇 가지 조건을 아래와 같이 작성하였지만, 막상 코드를 작성하지 못하고 있습니다.

조건

1. 새로운 페이지로 접속할 경우 prev 스택에 원래 있던 페이지를 넣고 next 스택을 비웁니다.
2. 뒤로 가기 버튼을 누를 경우 원래 있던 페이지를 next 스택에 넣고 prev 스택의 top에 있는 페이지로 이동한 뒤 prev 스택의 값을 pop 합니다.
3. 앞으로 가기 버튼을 누를 경우 원래 있던 페이지를 prev 스택에 넣고 next 스택의 top에 있는 페이지로 이동한 뒤 next 스택의 값을 pop 합니다.
4. 브라우저에서 뒤로 가기, 앞으로 가기 버튼이 비활성화일 경우(클릭이 되지 않을 경우)에는 스택에 push 하지 않습니다.

인터넷 브라우저에서 행동한 순서가 들어있는 배열 actions와 시작 페이지 start가 주어질 때 마지막에 접속해 있는 페이지와 방문했던 페이지들이 담긴 스택을 반환하는 솔루션을 만들어 김코딩의 궁금증을 풀어주세요.

입력

인자 1: actions

• String과 Number 타입을 요소로 갖는 브라우저에서 행동한 순서를 차례대로 나열한 배열

인자 2: start

• String 타입의 시작 페이지를 나타내는 현재 접속해 있는 대문자 알파벳

출력

• Array 타입을 리턴해야 합니다.

주의사항

• 만약 start의 인자로 string 자료형이 아닌 다른 자료형이 들어온다면 false를 리턴합니다.
• 새로운 페이지 접속은 알파벳 대문자로 표기합니다.
• 뒤로 가기 버튼을 누른 행동은 -1로 표기합니다.
• 앞으로 가기 버튼을 누른 행동은 1로 표기합니다.
• 다음 방문할 페이지는 항상 현재 페이지와 다른 페이지로 접속합니다.
• 방문한 페이지의 개수는 100개 이하입니다.
• 반환되는 출력값 배열의 첫 번째 요소 prev 스택, 세 번째 요소 next 스택은 배열입니다. 스택을 사용자 정의한다면 출력에서는 배열로 변환해야 합니다.

입출력 예시

const actions = ["B", "C", -1, "D", "A", -1, 1, -1, -1];
const start = "A";
const output = browserStack(actions, start);

console.log(output); // [["A"], "B", ["A", "D"]]

const actions2 = ["B", -1, "B", "A", "C", -1, -1, "D", -1, 1, "E", -1, -1, 1];
const start2 = "A";
const output2 = browserStack(actions2, start2);

console.log(output2); // [["A", "B"], "D", ["E"]]

 

 

 

작성 코드

function browserStack(actions, start) {
  // TODO: 여기에 코드를 작성합니다.
  if (typeof start !== 'string')
    return false;

  let prev = [];
  let curr = start;
  let next = [];

  for (let i = 0; i < actions.length; i++) {
    if (actions[i] === -1 && prev.length > 0) { // 뒤로가기 버튼이 눌리지 않을 때는 수행X
      next.push(curr);
      curr = prev.pop();
    }
    if (actions[i] === 1 && next.length > 0) {
      prev.push(curr);
      curr = next.pop();
    }
    else if (typeof actions[i] === "string") {   // 새로운 페이지 접속시에만 스택에 넣어야 하므로 
      prev.push(curr);                           // 해당 요소가 페이지 접속(알파벳 타입)일 때만 
      curr = actions[i];                         // 수행할 수 있게 조건문 추가!
      next = [];
    }
  }

  return [prev, curr, next];
}

 

Reference Code

function browserStack(actions, start) {
  // start 인자가 string이 아닌 것들은 전부 false로 리턴합니다.
  if(typeof start !== 'string') return false;

  // 뒤로 가기와 앞으로 가기 스택의 변수를 설정합니다
  let prevStack = [];
  let nextStack = [];
  let current = start;
  
  // actions 배열을 전부 탐색하기 위해 반복문을 설정합니다.
  for(let i = 0; i < actions.length; i++) {
    // 만약 actions 배열의 요소가 -1이고(뒤로가기를 눌렀고), prevStack의 길이가 0이 아닐 때(이전으로 돌아가는 페이지가 있다면)
    if(actions[i] === -1 && prevStack.length !== 0) {

      // prevStack에서 pop한 요소를 prevPage로 할당합니다.
      // nextStack에 current를 삽입합니다.
      // current를 prevPage에 할당합니다.
      let prevPage = prevStack.pop();
      nextStack.push(current);
      current = prevPage;

      // 만약 actions 배열의 요소가 1이고(앞으로가기를 눌렀고), nextStack의 길이가 0이 아닐 때(다음으로 넘어가는 페이지가 있다면)
    } else if(actions[i] === 1 && nextStack.length !== 0) {

      // nextStack에서 pop한 요소를 nextPage로 할당합니다.
      // prevStack에 current를 삽입합니다.
      // current를 nextPage에 할당합니다.
      let nextPage = nextStack.pop();
      prevStack.push(current);
      current = nextPage;

      // 만약 actions 배열의 요소가 알파벳이라면 (새로운 페이지라면)
    } else if(typeof actions[i] === 'string') {
      
      // prevStack에 current를 삽입합니다.
      // current에 현재 알파벳 요소를 할당합니다.
      // 새로운 페이지는 앞으로 갈 수 없기 때문에 nextStack을 비웁니다.
      prevStack.push(current);
      current = actions[i];
      nextStack = [];
    }

    // 이외엔, 동작하지 않습니다.
  }
  
  // 배열에 prevStack, current, nextStack을 순서대로 담아 반환합니다.
  return [prevStack, current, nextStack];
}

 

---

 

문제 조건에서 친절하게 step을 알려주기때문에, 스택의 기본 원리만 잘 이해하고 있다면 그리 어렵지 않은 문제였던 것 같다.