C++ / 영어 끝말잇기

Problem

문제 난이도 : Programmers 2단계 69%

분류 : Implementation 같음

Solution

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;

vector<int> solution(int n, vector<string> words) {
    vector<int> answer;
    answer.push_back(0);
    answer.push_back(0);
    vector<string> UsedWords;
    UsedWords.push_back(words[0]);
    for(int i = 1; i < words.size(); i++){
        if(find(UsedWords.begin(),UsedWords.end(),words[i]) == UsedWords.end()){
           if(UsedWords.back().back() != words[i].front()){
               cout << i << endl;
               answer[0] = i % n + 1;
               answer[1] = i / n + 1;
               break;
           }else{
               UsedWords.push_back(words[i]);
           }
        }else{
            answer[0] = i % n + 1;
            answer[1] = i / n + 1;
            cout << i << endl;
            break;
        }
    }
    
    return answer;
}