영어 끝말잇기/C++

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

프로그래머스

풀이 방법

1. 같은 단어를 말하는지 검사하기 위해 string을 Key, int를 Value로 가지는 map형을 선언

2. 첫 단어를 삽입하고 m[words[i]]++해준다.

3. 이후부터는 앞의 단어의 끝 문자와 현재 단어의 첫 문자가 다르거나, m[words[i]]가 1이면 해당 순서의 번호와 차례를  리턴한다.

4. 위의 조건에 만족하지 않는다면 \m[words[i]]++을 하고 계속해서 반복한다.

5. 만약 탈락자가 생기지 않는다면 {0, 0}을 리턴

#include <string>
#include <vector>
#include <map>

using namespace std;

vector<int> solution(int n, vector<string> words) {
    map<string, int> m;
    
    for(int i = 0; i < words.size(); i++) {
        if(i > 0) {
            if(words[i].front() != words[i-1].back() || m[words[i]])
                return {i % n + 1, i / n + 1};
        }
        m[words[i]]++;
    }

    return {0,0};
}