[프로그래머스] Lv.2 영어 끝말잇기 - 자바스크립트 풀이

문제

문제 링크

설명

반복문을 돌면서 단어를 집합에 넣는다.

이미 집합에 있는 단어이거나 이전 단어의 끝 문자와 현재 단어의 첫 문자가 다르면 계산해서 반환한다.

반복문을 다 돌았다면 [0, 0]을 반환한다.

코드

function solution(n, words) {
  let s = new Set();
  s.add(words[0]);
  for (let i = 1; i < words.length; i++) {
    let prev = words[i - 1];
    let curr = words[i];
    if (s.has(curr) || prev.at(-1) !== curr.at(0)) {
      return [(i % n) + 1, parseInt(i / n) + 1];
    } else s.add(curr);
  }
  return [0, 0];
}