모의고사
문제 설명
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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 |