모의고사
문제 설명
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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 | return |
[1, 2, 3, 4, 5] | [1] |
[1, 3, 2, 4, 2] | [1, 2, 3] |
Python 코드
def solution(answers):
answer = []
# 패턴정의
first = [ 1,2,3,4,5 ]
second = [ 2,1,2,3,2,4,2,5 ]
third = [ 3,3,1,1,2,2,4,4,5,5 ]
# 점수정의
first_count = 0
second_count = 0
third_count = 0
# 정답확인
for number in range(len(answers)):
if answers[ number ] == first[ number % 5 ]:
first_count += 1
if answers[ number ] == second[ number % 8 ]:
second_count += 1
if answers[ number ] == third[ number %10 ]:
third_count += 1
pre_answer = [ first_count,second_count,third_count ]
# 가장 많이 맞힌 사람
for person, score in enumerate(pre_answer):
if score == max(pre_answer):
answer.append(person + 1)
return answer
C++ 코드
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
int test1[5] = {1, 2, 3, 4, 5};
int test2[8] = {2, 1, 2, 3, 2, 4, 2, 5};
int test3[10] = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};
vector<int> solution(vector<int> answers) {
vector<int> answer;
int score[3] = {0, };
int max_score = 0;
for (int i = 0; i < answers.size(); i++){
if (answers[i] == test1[i % 5]) score[0] += 1;
if (answers[i] == test2[i % 8]) score[1] += 1;
if (answers[i] == test3[i % 10]) score[2] += 1;
}
max_score = max(max(score[0], score[1]), score[2]);
for (int i = 0; i < 3; i++){
if (score[i] == max_score)
answer.push_back(i + 1);
}
return answer;
}
* 참고 링크 : https://rile1036.tistory.com/28
출처
'코딩테스트 > Programmers' 카테고리의 다른 글
[Leet Code] 374. Guess Number Higher or Lower (0) | 2022.01.28 |
---|---|
[코딩테스트/Programmers] 25_소수찾기 (Lv2.) (0) | 2022.01.25 |
[코딩테스트/Programmers] 23_주식가격 (Lv2.) (0) | 2022.01.18 |
[코딩테스트/Programmers] 22_기능개발 (Lv2.) (0) | 2022.01.17 |
[코딩테스트/Programmers] 21_다리를 지나는 트럭 (Lv2.) (0) | 2022.01.17 |