[프로그래머스/Java] 모의고사

https://programmers.co.kr/learn/courses/30/lessons/42840?language=java 

코딩테스트 연습 - 모의고사

 

import java.util.*;
class Solution {
    public int[] solution(int[] answers) {
        int[] a = {1,2,3,4,5};
        int[] b = {2,1,2,3,2,4,2,5};
        int[] c = {3,3,1,1,2,2,4,4,5,5};
        int result1=0, result2=0, result3=0;
        
        for(int i=0; i<answers.length; i++){
            if(answers[i]==a[i%a.length]) result1++;
            if(answers[i]==b[i%b.length]) result2++;
            if(answers[i]==c[i%c.length]) result3++;
        }
        
        int max = Math.max(Math.max(result1, result2),result3);
        ArrayList<Integer> list = new ArrayList<Integer>();
        if(max == result1) list.add(1);
        if(max == result2) list.add(2);
        if(max == result3) list.add(3);
        
        int[] answer = new int[list.size()];
        
        for(int i=0; i<answer.length; i++){
            answer[i] = list.get(i);
        }
        
        return answer;
    }
}

내가 생각 못한 점

 

1. 각각 맞은 개수 카운트 하는 부분

방식은 같았지만 나는 한사람씩 함수를 만들어서 리턴했는데 이렇게 하니까 간단하게 카운트가 되었다.

for(int i=0; i<answers.length; i++){
      if(answers[i]==a[i%a.length]) result1++;
      if(answers[i]==b[i%b.length]) result2++;
      if(answers[i]==c[i%c.length]) result3++;
}

 

2. 맞은 개수들 중 최대값 구하는 방법

나는 두개씩 비교하면서 최대값을 구했는데 간단한 Math.max()라는 메서드가 있었다.

int max = Math.max(Math.max(result1, result2),result3);

 

3. 맞은 개수가 최대로 같은게 몇 개 일 줄 알고 answer[]을 선언하지 했는데

ArrayList<Integer> 에 먼저 담고 이 리스트의 크기로 선언하면 됐다.

크기가 고정되는 배열에 반해 ArrayList는 가변적이라 그런 것 같다.

ArrayList<Integer> list = new ArrayList<Integer>();
if(max == result1) list.add(1);
if(max == result2) list.add(2);
if(max == result3) list.add(3);

int[] answer = new int[list.size()];

 

넣을 때는 하나씩

for(int i=0; i<answer.length; i++){
    answer[i] = list.get(i);
}

 

 

출처 :

https://sas-study.tistory.com/240