본문 바로가기

algorithm

[프로그래머스]영어 끝말잇기

https://programmers.co.kr/learn/courses/30/lessons/12981

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

문제

사람의 수 n과 사람들이 순서대로 말한 단어 words가 매개변수로 주어질 때, 가장 먼저 탈락하는 사람의 번호와 그 사람들이 자신이 몇 번째 차례에 탈락하는지를 구해서 return 하도록 solution 함수를 완성해주세요

 

제한사항

  • 끝말잇기에 참여하는 사람의 수 n은 2 이상 10 이하의 자연수 입니다.
  • words는 끝말잇기에 사용한 단어들이 순서대로 들어있는 배열이며, 길이는 n 이상 100 이하입니다.
  • 단어 길이는 2 이상 50 이하이다.
  • 모든 단어는 알파벳 소문자로만 이러우져 있습니다.
  • 끝말잇기에 사용되는 단어의 뜻(의미)은 신경 쓰이지 않으셔도 됩니다.
  • 정단은 [번호, 차례] 형태로 return 해주세요
  • 만약 주어진 단어들로 탈락자가 생기지 않는다면, [0, 0]을 return 해주세요.

입출력의 예

 

풀이

 

ArrayList에 단어를 넣어서 같은 단어가 있는지 파악하였고, 

arr 배열을 통해 n번 째 사람의 끝말잇기 차례를 파악하기위해서 선언하였습니다. 

    ArrayList<String> list = new ArrayList<>();
    int[] arr = new int[n]; 

 

i % n(n을 통해 몇 사람이 끝말잇기를 하는지 알 수 있다.)을 통해 몇 번째 사람인지 구하고, ++를 통해 그 사람이 몇 번째 차례인지 구합니다.

indexOf를 통해 list 앞에 중복된 단어가 있는지 확인하고 없으면 값을 넣어주고, 있으면 끝말 잇기가 끝나야 하기 때문에 (i % n + 1)을 통해 몇 번째 사람인지 구하고, 

arr[i % n]을 통해 틀린 사람이 몇 번째 차례인지 구한다.

    arr[i % n]++;			
    
    if(list.indexOf(words[i]) == -1) {
        list.add(words[i]);
    }else {
        answer[0] = i % n + 1;
        answer[1] = arr[i % n];
        return answer;
    }

 

i > 0 이라는 뜻은 list에 값이 하나 이상 들어가 있다는 뜻으로, 앞 단어의 제일 마지막 알파벳과 뒷 단어의 제일 앞 알파벳같은지 파악하기 위해 조건을 걸었습니다.

    if (i > 0) {
        if (list.get(i - 1).charAt(words[i - 1].length() - 1) != list.get(i).charAt(0)) {
            answer[0] = i % n + 1;
            answer[1] = arr[i % n];
            return answer;            
        }
    }

 

위에 두 조건을 무사히 통과하면 끝말 잇기가 잘 끝났다는 의미로 각, 각 배열에 '0'을 대입합니다.

 

    answer[0] = 0;
    answer[1] = 0;