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

코딩테스트 연습 - 영어 끝말잇기

 

문제설명

제한사항

입출력예

 

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

vector<int> solution(int n, vector<string> words) {
	int size, who, ord;
	string first = "", last = "";
	vector<int> answer;
	unordered_map<string, int>cache;

	for (string w : words) {

		cache[w]++;
		first = w[0];

		if (cache.size() != 1) {
			if (cache[w]>1 || last != first) {
				size = (1 < cache[w]) ?  cache.size() + 1 : cache.size();
				who = (size % n == 0) ? n : size % n;
				ord = (size%n == 0) ? size / n : size / n + 1;
				answer.push_back(who);
				answer.push_back(ord);
				return answer;
			}
		}
		last = w[w.size() - 1];
	}
	answer.push_back(0);
	answer.push_back(0);
	return answer;
}