mun dev

[COS PRO 1급] 1-8 누가 당선 되나요 자바(Java) 본문

알고리즘/COS PRO 1급

[COS PRO 1급] 1-8 누가 당선 되나요 자바(Java)

mndev 2023. 11. 25. 16:18

문제링크

 

구름HOME

구름은 클라우드 기술을 이용하여 누구나 코딩을 배우고, 실력을 평가하고, 소프트웨어를 개발할 수 있는 클라우드 소프트웨어 생태계입니다.

www.goorm.io

 

문제유형

한줄 바꾸기 문제

 

문제

import java.util.*;
 
class Main {
    
    public int[] solution(int N, int[] votes) {
        int voteCounter[] = new int[11];
        for (int i = 0; i < votes.length; i++) {
            voteCounter[votes[i]] += 1;
        }
        int maxVal = 0;
        int cnt = 0;
        for (int i = 1; i <= N; i++) {
            if (maxVal < voteCounter[i]) {
                maxVal = voteCounter[i];
                cnt = 1;
            }
            else if(maxVal == voteCounter[i]){
                cnt += 1;
            }
        }
        int answer[] = new int[cnt];
        for (int i = 1, idx = 0; i <= N; i++){
            if (voteCounter[i] == maxVal) {
                answer[idx] = voteCounter[i];
                idx += 1;
            }
        }
            Arrays.sort(answer);
        return answer;
    }
    
public static void main(String[] args) {
        Main sol = new Main();
        int N1 = 5;
        int[] votes1 = {1,5,4,3,2,5,2,5,5,4};
        int[] ret1 = sol.solution(N1, votes1);
 
        System.out.println("solution 함수의 반환 값은 " + Arrays.toString(ret1) + " 입니다.");
        
 
        int N2 = 4;
        int[] votes2 = {1, 3, 2, 3, 2};
        int[] ret2 = sol.solution(N2, votes2);
 
        System.out.println("solution 함수의 반환 값은 " + Arrays.toString(ret2) + " 입니다.");
    }
}

 

문제 풀이

import java.util.*;
 
class Main {
    
    public int[] solution(int N, int[] votes) {
        int voteCounter[] = new int[11];
        for (int i = 0; i < votes.length; i++) {
            voteCounter[votes[i]] += 1;
        }
        int maxVal = 0;
        int cnt = 0;
        for (int i = 1; i <= N; i++) {
            if (maxVal < voteCounter[i]) {
                maxVal = voteCounter[i];
                cnt = 1;
            }
            else if(maxVal == voteCounter[i]){
                cnt += 1;
            }
        }
        int answer[] = new int[cnt];
        for (int i = 1, idx = 0; i <= N; i++){
            if (voteCounter[i] == maxVal) {
                answer[idx] = i;
                idx += 1;
            }
        }
            Arrays.sort(answer);
        return answer;
    }
    
public static void main(String[] args) {
        Main sol = new Main();
        int N1 = 5;
        int[] votes1 = {1,5,4,3,2,5,2,5,5,4};
        int[] ret1 = sol.solution(N1, votes1);
 
        System.out.println("solution 함수의 반환 값은 " + Arrays.toString(ret1) + " 입니다.");
        
 
        int N2 = 4;
        int[] votes2 = {1, 3, 2, 3, 2};
        int[] ret2 = sol.solution(N2, votes2);
 
        System.out.println("solution 함수의 반환 값은 " + Arrays.toString(ret2) + " 입니다.");
    }
}