https://school.programmers.co.kr/learn/courses/30/lessons/42840?language=python3
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
c++로 풀어져 있었던 문제 . .
python 으로 다시 풀었는데 비슷하게 풀었다.
def solution(answers):
arr1 = [1,2,3,4,5]
arr2 = [2,1,2,3,2,4,2,5]
arr3 = [3,3,1,1,2,2,4,4,5,5]
n1,n2,n3 = 0,0,0
for i in range(len(answers)):
n1 += (answers[i] == arr1[i%len(arr1)])
n2 += (answers[i] == arr2[i%len(arr2)])
n3 += (answers[i] == arr3[i%len(arr3)])
cnt = [n1,n2,n3]
return [i+1 for i in range(3) if max(cnt) == cnt[i]]
#include <string>
#include <vector>
using namespace std;
vector<int> solution(vector<int> answers) {
vector<int> answer;
vector<int> answer1 {1,2,3,4,5};
vector<int> answer2 {2,1,2,3,2,4,2,5};
vector<int> answer3 {3,3,1,1,2,2,4,4,5,5};
int count1 = 0, count2 = 0, count3 = 0;
for (int i = 0; i < answers.size(); ++i){
if (answers[i] == answer1[i % 5]) count1++;
if (answers[i] == answer2[i % 8]) count2++;
if (answers[i] == answer3[i % 10]) count3++;
}
int maxCount = max(count1, max(count2, count3));
if (maxCount == count1) answer.push_back(1);
if (maxCount == count2) answer.push_back(2);
if (maxCount == count3) answer.push_back(3);
return answer;
}