C++ 모의고사 - Programmers

코딩테스트 연습 모의고사

---

모의고사

문제 설명

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.

1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, … 2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, … 3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, …

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한 사항

• 시험은 최대 10,000 문제로 구성되어있습니다.
• 문제의 정답은 1, 2, 3, 4, 5 중 하나입니다.
• 가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

입출력 예

answers = [1,2,3,4,5]
return = [1]

입출력 예 설명

입출력 예 #1

• 수포자 1은 모든 문제를 맞혔습니다.
• 수포자 2는 모든 문제를 틀렸습니다.
• 수포자 3은 모든 문제를 틀렸습니다. 따라서 가장 문제를 많이 맞힌 사람은 수포자 1입니다.

코드

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

void check_end(vector<int>::iterator& iter, vector<int>& vector);
// 배열의 끝이여서 처음으로 돌아가야하는지 확인하는 함수

void compare_number(vector<int>& answer, int count[]);
// 많이 맞은 갯수 찾고 그 값을 가진 사람 vector에 담는 함수

vector<int> solution(vector<int> answers) {
  vector<int> answer;

  vector<int> numbers1 = { 1, 2, 3, 4, 5 };
	vector<int> numbers2 = { 2, 1, 2, 3, 2, 4, 2, 5};
	vector<int> numbers3 = { 3, 3, 1, 1, 2, 2, 4, 4, 5, 5 };
	vector<int>::iterator it1 = numbers1.begin();
	vector<int>::iterator it2 = numbers2.begin();
	vector<int>::iterator it3 = numbers3.begin();

  int answers_size = answers.size();

  int count[3] = { 0 };	// 각 사람이 맞춘 개수

	for (int i = 0; i < answers_size; i++)
	{
		if (answers[i] == *it1)	count[0]++; // 정답 확인
		if (answers[i] == *it2)	count[1]++;
		if (answers[i] == *it3)	count[2]++;
		it1++, it2++, it3++;
		check_end(it1, numbers1);
		check_end(it2, numbers2);
		check_end(it3, numbers3);
	}
  compare_number(answer, count);

  return answer;
}

void check_end(vector<int>::iterator& iter, vector<int>& vector)
{
	if (iter == vector.end())
		iter = vector.begin();
}

void compare_number(vector<int>& answer, int count[])
{
	int max = 0;
	for (int i = 0; i < 3; i++)	// max 값 찾기
	{
		if (max <= count[i])
			max = count[i];
	}
	for (int i = 0; i < 3; i++)	// max 값인 index찾고 answer에 담기
	{
		if (max == count[i])
			answer.push_back(i+1);
	}
	// sort
	sort(answer.begin(), answer.end());
}

고찰

linked list를 활용해서 문제를 풀려고도 고민했었지만, vector를 이용해 푸는 것이 훨씬 간단하다고 생각해서 vector를 사용해 풀었다. linked list를 이용해서도 쉽게 풀 수 있을 것 같다.