[프로그래머스] Lv.1 과일장수

문제 : Lv.1 과일장수 https://school.programmers.co.kr/learn/courses/30/lessons/135808 언어 : JAVA 체감 Level : ★ ☆ ☆ Review: 편하게 풀어보려고 컬렉션으로 정렬해서 풀어봤는데, 역루프 문으로 충분히 깔끔하게 짤 수 있음을 깨달았다. 무작정 편하려고 컬렉션 사용하는게 아닌 핵심 로직을 잘 이해하도록하자

 

 

💡

import java.util.*;
import java.util.stream.*;

class Solution {
    public int solution(int k, int m, int[] score) {

        // stream 으로 배열 -> 리스트로 변환 -> 내림차순
        List<Integer> score_list = Arrays.stream(score)
                                 .boxed()         // int → Integer
                                  .sorted(Comparator.reverseOrder())
                                 .collect(Collectors.toList());
        
        // 큐 생성
        Queue<Integer> queue = new LinkedList<>();
        score_list.forEach(queue::offer);

        
        // 최대 바구니 값
        int max_cnt = score.length / m ;         
        int cnt = 0;
        int answer = 0;
        
        
        while (true){ 
            if(cnt == max_cnt) { 
                break;
            }
                            
            for(int i = 1; i <= m ; i++){ // 최대 사과 갯수만큼 꺼내기
                if (i == m){  //  이때 바구니 안의 최저값만 
                    answer += queue.poll() * m;
                } else {
                     queue.poll();
                }
            }
            cnt ++;         
        }
    
        
        return answer;
    }
}

 

 

💡 다른 방법 

 

1. 오름차순 정렬

2. 뒤에서부터 4개씩 끊어서 작업해준다.

 

import java.util.*;

class Solution {
    public int solution(int k, int m, int[] score) {
        int answer = 0;

        Arrays.sort(score);

        for(int i = score.length; i >= m; i -= m){
            answer += score[i - m] * m;
        }

        return answer;
    }
}