(C#) 베스트앨범

코딩테스트 연습 - 베스트앨범 | 프로그래머스 스쿨

프로그래머스

뭐 특별한 알고리즘이나 자료구조를 사용할 것도 없이 요구사항만 구현하면 된다.

 

입력받은 두 배열로 Dictionary를 만든다.

Dictionary는 장르와 해당 장르의 총합, 제일 많이 플레이된 노래, 두번째로 많이 플레이된 노래를 가진다.

다음은 장르들을 총합으로 정렬하고 순서대로 answer에 추가해준다.

 

answer는 배열 형태인데, 배열의 크기를 미리 정할 수 없어서 List<int> 형태로 먼저 만들고 array로 변환했다.

using System;
using System.Collections.Generic;
using System.Linq;

public class Solution {
    public int[] solution(string[] genres, int[] plays)
    {
        int[] answer = new int[] { };
        Dictionary<string, (int total, int first, int second)> map = new Dictionary<string, (int total, int first, int second)>();
        for (var i = 0; i < genres.Length; i++)
        {
            if (map.ContainsKey(genres[i]))
            {
                var value = map[genres[i]];
                value.total += plays[i];
                if (plays[i] > plays[value.first])
                {
                    value.second = value.first;
                    value.first = i;
                }
                else if (value.second == -1 || plays[i] > plays[value.second])
                {
                    value.second = i;
                }
                map[genres[i]] = value;
            }
            else
            {
                map.Add(genres[i], (plays[i], i, -1));
            }
        }
        List<(int total, int first, int second)> values = map.Values.ToList();
        values.Sort((l, r) =>
        {
            return r.total.CompareTo(l.total);
        });
        List<int> temp = new List<int>();
        for (var i = 0; i < values.Count; i++)
        {
            temp.Add(values[i].first);
            if (values[i].second != -1)
                temp.Add(values[i].second);
        }
        answer = temp.ToArray();

        return answer;
    }
}