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

문제 설명

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

제한사항

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

---

풀이

10분
string vector를 선언하여 앞에서 말한 단어인지 체크하고, 앞 단어와 이어지는지 체크하여 탈락자를 구했다.

코드

// 11:11 ~ 11:22
#include <string>
#include <vector>
#include <algorithm>
#include <iostream>

using namespace std;

vector<int> solution(int n, vector<string> words) {
    vector<int> answer(2,0);
    vector<string> voca;
    int i;
    for(i = 0; i < words.size(); ++i){
        if(i != 0 && words[i -1][words[i- 1].length() - 1] != words[i][0]) break;
        else if(find(voca.begin(),voca.end(),words[i]) != voca.end()) break;
        else voca.push_back(words[i]);
    }
    if(i == words.size()) return answer;
    answer[0] = i % n + 1;
    answer[1] = i / n + 1;
    return answer;
}