プログラマー完全ナビゲーション1シミュレーション試験

問題の説明

せいげんじょうけん

I/O例

に答える

class Solution {
    public int[] solution(int[] answers) {
        int[] a1 ={1,2,3,4,5};
        int[] a2 ={2, 1, 2, 3, 2, 4, 2, 5};
        int[] a3 = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};
        int c1=0;
        int c2=0;
        int c3=0;
        for(int i = 0; i<answers.length;i++){
            int a = answers[i];
            if (a1[(i%5)]==a) c1++;
            if (a2[(i%8)]==a) c2++;
            if (a3[(i%10)]==a) c3++;
        } 
        int[] answer = {};
        if(c1==c2&&c1==c3) return new int[] {1,2,3};
        if(c1>c2&&c1>c3) return new int[] {1};
        if(c2>c1&&c2>c3) return new int[] {2};
        if(c3>c2&&c3>c1) return new int[] {3};
        if(c1==c2&&c1>c3) return new int[] {1,2};
        if(c1==c3&&c1>c2) return new int[] {1,3};         
        if(c2==c3&&c2>c1) return new int[] {2,3};    
        return answer;
    }
}