[프로그래머스, Java] 튜플

 

1. 문자열 구분이 필요 "{" "," "}" 별로 수행해야하는 로직을 정의해서 구현

2. 주어지는 값이 중복이 없고, 길이도 제각각 (a3, a2, a4, a1) 요런식으로 다르게 주어지므로 List<List<Integer>> 선택

3. 하지만 List<List<Integer>>를 listSize별로 정렬할건데 성능 개선을 위해 연산을 하면서 listSize를 미리 증가 연산을 해서 리스트 정렬하기 위해 리스트 전체를 카운트하지 않도록 구현

4. 이를 위해 List, 리스트크기를 담는 Element 클래스를 작성

5. 이후는 코드 내용과 그대로 진행

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

class Solution {
    public int[] solution(String s) {
        int[] answer = {2, 1, 3, 4};
        int sLen = s.length();

        List<Element> numSet = new ArrayList<>();
        List<Integer> list = new ArrayList<>();
        StringBuilder sb = new StringBuilder();
        int listSize = 0;

        for (int i = 1; i < sLen - 1; i++) {
            if (s.charAt(i) == '{') {
                list = new ArrayList<>();
                listSize = 0;
                sb = new StringBuilder();
            } else if (s.charAt(i) == ',') {
                if (s.charAt(i - 1) == '}')
                    continue;
                list.add(Integer.valueOf(sb.toString()));
                sb = new StringBuilder();
                listSize++;
            } else if (s.charAt(i) == '}') {
                list.add(Integer.valueOf(sb.toString()));
                sb = new StringBuilder();
                listSize++;
                numSet.add(new Element(list, listSize));
            } else {
                sb.append(s.charAt(i));
            }
        }

        numSet.sort((Element e1, Element e2) -> e1.listSize - e2.listSize);
        Map<Integer, Integer> checkMap = new HashMap<>();
        List<Integer> result = new ArrayList<>();

        for (Element element : numSet) {
            element.list.forEach(e -> {
                if (checkMap.getOrDefault(e, -1) == -1) {
                    checkMap.put(e, 1);
                    result.add(e);
                }
            });
        }

        int resultSize = result.size();
        answer = new int[resultSize];

        for (int i = 0; i < resultSize; i++) {
            answer[i] = result.get(i);
        }

        return answer;
    }

    class Element {
        List<Integer> list = new ArrayList<>();
        int listSize = 0;

        public Element(List<Integer> list, int listSize) {
            this.list = list;
            this.listSize = listSize;
        }
    }
}