개발 다이어리

문제

https://school.programmers.co.kr/learn/courses/30/lessons/42579

문제 설명

코드

import java.util.*;
import java.util.stream.IntStream;
import java.util.stream.Stream;

import static java.util.stream.Collectors.*;

class Solution {
	static int i = 0;

	public int[] solution(String[] genres, int[] plays) {
		return IntStream.range(0, genres.length) // intStream
				        .mapToObj(i-> new TopGenrePlay(i, genres[i], plays[i])) // 기본형에서 참조형으로
				        .collect(groupingBy( TopGenrePlay::getGenre))			 // key는 genres[i] 객체 그대로 Map<String, List<TopGenrePlay>>
				        .entrySet().stream()	//   Stream<Entry<String, List<TopGenrePlay>>>
				        .sorted( (a,b)-> { // 스트림 안에서 총 Play집계 값으로 내림차순
					       	 int sum1 = a.getValue().stream().mapToInt(TopGenrePlay::getPlay).sum();
					       	 int sum2 = b.getValue().stream().mapToInt(TopGenrePlay::getPlay).sum();
					       	 return Integer.compare(sum2, sum1);}) //내림차순  
				        .map( entry -> { //<Entry<String, List<TopGenrePlay>>> 
					       	 List tgpList = entry.getValue().stream().sorted()//정렬은 객체에 구현된 Comparable에 의존한다.
							       			 		                 .limit(2)//정렬된 결과에서 2개만 가져온다.
							       			 		                 .map( TopGenrePlay::getId ).collect(toList());//식별값만 필요하다.
					       	 entry.setValue(tgpList);
					       	 return entry; } )
				        .flatMap( entry-> Stream.of(entry.getValue().toArray(new Integer[] {}))) // 하나의 스트림으로 만든다.
				        .mapToInt( Integer::intValue ) //기본형 스트림으로 IntStream
				        .toArray()
				        ;
	}

	public static void main(String[] args) {
		Solution s = new Solution();
		System.out.println(Arrays.toString(s.solution(
				new String[] { "classic", "pop", "classic", "classic", "pop", "trot", "trot", "trot", "pop" },
				new int[] { 1000, 600, 2000, 1000, 2500, 6000, 200, 200, 600 })));

	}

	class TopGenrePlay implements Comparable<TopGenrePlay> {
		int id;
		String genre;
		int play;

		public int getId() {
			return id;
		}

		public void setId(int id) {
			this.id = id;
		}

		public String getGenre() {
			return genre;
		}

		public void setGenre(String genre) {
			this.genre = genre;
		}

		public int getPlay() {
			return play;
		}

		public void setPlay(int play) {
			this.play = play;
		}

		public TopGenrePlay(int id, String genre, int play) {
			super();
			this.id = id;
			this.genre = genre;
			this.play = play;
		}
		@Override
		public String toString() {
			return "[id="+ id + ", genre=" + genre + ", play=" + play + "]";
		}
		@Override
		public int compareTo(TopGenrePlay o) {
			int result = Integer.compare(o.play, this.play);
			if (result == 0) {
				return Integer.compare(this.id, o.id);
			}
			return result;
		}
	}

}

결과

정확성  테스트
테스트 1 〉	통과 (17.71ms, 82.3MB)
테스트 2 〉	통과 (13.47ms, 77.6MB)
테스트 3 〉	통과 (15.14ms, 81.7MB)
테스트 4 〉	통과 (15.90ms, 73.1MB)
테스트 5 〉	통과 (9.80ms, 81.5MB)
테스트 6 〉	통과 (15.30ms, 73.7MB)
테스트 7 〉	통과 (10.73ms, 72.7MB)
테스트 8 〉	통과 (9.44ms, 73.7MB)
테스트 9 〉	통과 (9.91ms, 73.8MB)
테스트 10 〉	통과 (16.23ms, 80.8MB)
테스트 11 〉	통과 (8.91ms, 75MB)
테스트 12 〉	통과 (8.83ms, 75.1MB)
테스트 13 〉	통과 (14.77ms, 81.3MB)
테스트 14 〉	통과 (10.05ms, 78.5MB)
테스트 15 〉	통과 (8.19ms, 75.4MB)

퇴근하고 이틀 동안 계속 붙잡았다... 컬렉션으로만 해결하려 했던게 오히려 나에겐 독이였다.

될듯..?말듯...

그러다 문득, 내가 자바를 하는데 너무 함수형으로 생각한 게 아닌가 라는 생각이 들었다.

갑자기 Class를 활용할 생각을 안하고 컬렉션만 쓰려했는지...참...

그래도 이 교훈은 오래 남을 듯 하다.

누군가 봐서 조금이라도 도움이 됐으면하는 마음으로 중간중간에 데이터 형을 입력했다.

위 코드의 개략적인 구도는 play, id 순 정렬은 객체의 Comparable 구현에 의존한다.

이 정렬 전에 장르별 정렬은 스트림을 사용해 장르별 그룹화 후 

내부 List 반복으로 총 집계를 내어 정렬했다.

필요한 정렬정보는 모두 모였으므로, 데이터 가공으로 id 배열을 만들어 리턴했다.