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

문제

---

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

풀이

---

이전에 나온 단어가 나오면 안되므로 map을 사용하여 중복된 값이 나왔을 때 해당 번호와 사이클을 리턴하고, 중복된 값이 없을 경우 (0,0)을 리턴해준다.

#include <string>
#include <vector>
#include <iostream>
#include <unordered_map>

using namespace std;

vector<int> solution(int n, vector<string> words) 
{
    vector<int> answer;

    unordered_map <string, int> m;

    int i = 0;
    for (; i < words.size(); i++)
    {
        m.insert({ words[i], i % n + 1});

        if (m.size() != i + 1)
        {
            answer.push_back(i % n + 1);
            answer.push_back(i / n + 1);
            break;
        }

        if (i >= 1)
        {
            if (words[i - 1][words[i - 1].size()-1] != words[i][0])
            {
                answer.push_back(i % n + 1);
                answer.push_back(i / n + 1);
                break;
            }
             
        }
    }

    if (i == words.size())
    {
        answer.push_back(0);
        answer.push_back(0);
    }

    return answer;
}