상세 컨텐츠

본문 제목

2022년 4월 12일 - 튜플 lv2

컴퓨터 공부/코딩테스트 스터디

by 주중 (zuzung) 2022. 4. 20. 10:12

본문

문제 설명

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(tuple)이라고 합니다. n개의 요소를 가진 튜플을 n-튜플(n-tuple)이라고 하며, 다음과 같이 표현할 수 있습니다.

  • (a1, a2, a3, ..., an)

튜플은 다음과 같은 성질을 가지고 있습니다.

  1. 중복된 원소가 있을 수 있습니다. ex : (2, 3, 1, 2)
  2. 원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플입니다. ex : (1, 2, 3) ≠ (1, 3, 2)
  3. 튜플의 원소 개수는 유한합니다.

원소의 개수가 n개이고, 중복되는 원소가 없는 튜플 (a1, a2, a3, ..., an)이 주어질 때(단, a1, a2, ..., an은 자연수), 이는 다음과 같이 집합 기호 '{', '}'를 이용해 표현할 수 있습니다.

  • {{a1}, {a1, a2}, {a1, a2, a3}, {a1, a2, a3, a4}, ... {a1, a2, a3, a4, ..., an}}

예를 들어 튜플이 (2, 1, 3, 4)인 경우 이는

  • {{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}

와 같이 표현할 수 있습니다. 이때, 집합은 원소의 순서가 바뀌어도 상관없으므로

  • {{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}
  • {{2, 1, 3, 4}, {2}, {2, 1, 3}, {2, 1}}
  • {{1, 2, 3}, {2, 1}, {1, 2, 4, 3}, {2}}

는 모두 같은 튜플 (2, 1, 3, 4)를 나타냅니다.

특정 튜플을 표현하는 집합이 담긴 문자열 s가 매개변수로 주어질 때, s가 표현하는 튜플을 배열에 담아 return 하도록 solution 함수를 완성해주세요.

[제한사항]

  • s의 길이는 5 이상 1,000,000 이하입니다.
  • s는 숫자와 '{', '}', ',' 로만 이루어져 있습니다.
  • 숫자가 0으로 시작하는 경우는 없습니다.
  • s는 항상 중복되는 원소가 없는 튜플을 올바르게 표현하고 있습니다.
  • s가 표현하는 튜플의 원소는 1 이상 100,000 이하인 자연수입니다.
  • return 하는 배열의 길이가 1 이상 500 이하인 경우만 입력으로 주어집니다.

[입출력 예]

해답

def solution(s):
    t = s.split("},")  //'},'를 기준으로 값을 나눈다.
    p = []  //이 곳은 나눠진 튜플들이 들어갈 배열이다.
    for i in range(len(t)):
        t[i] = t[i].replace("{","")  //불필요한 '{'값을 replace 함수를 이용하여 지운다.
        t[i] = t[i].replace("}","")  //불필요한 '}'값을 replace 함수를 이용하여 지운다.
        p.append(t[i].split(","))  //남아있는 값인 ','를 기준으로 배열로 만든다.
    p.sort(key=len)  //길이 순으로 정렬한다.
    answer = []
    for i in range(len(p)):
        for j in range(len(p[i])):
            if int(p[i][j]) not in answer:  //없는 값을 answer에 순서대로 삽입한다.
                answer.append(int(p[i][j]))
    return answer
[나의 풀이]
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;

class Solution {
    public int[] solution(String s) {
        String[] t = s.split("},");
        ArrayList<String[]> p = new ArrayList<>();
        for(int i=0; i<t.length; i++){
            t[i] = t[i].replace("{","");
            t[i] = t[i].replace("}","");
            p.add(t[i].split(","));
        }
        Collections.sort(p, new Comparator<String[]>() {
            public int compare(String[] o1, String[] o2) {
                if(o1.length < o2.length){
                    return -1;
                } else if(o1.length == o2.length){
                    return 0;
                } else {
                    return 1;
                }
            }
        });
        ArrayList<Integer> answer = new ArrayList<>();
        for(int i=0; i<p.size();i++){
            for(int j=0; j<p.get(i).length;j++){
                if (!answer.contains(Integer.valueOf(p.get(i)[j])))
                    answer.add(Integer.valueOf(p.get(i)[j]));
            }
        }
        return answer.stream().mapToInt(i -> i).toArray();
    }
}
[다른 사람의 풀이]
import java.util.*;
class Solution {
    public int[] solution(String s) {
        Set<String> set = new HashSet<>();
        String[] arr = s.replaceAll("[{]", " ").replaceAll("[}]", " ").trim().split(" , ");
        Arrays.sort(arr, (a, b)->{return a.length() - b.length();});
        int[] answer = new int[arr.length];
        int idx = 0;
        for(String s1 : arr) {
            for(String s2 : s1.split(",")) {
                if(set.add(s2)) answer[idx++] = Integer.parseInt(s2);
            }
        }
        return answer;
    }
}

느낀 점

notion 코드 적는 것은 너무 느려서 기존의 코드에 주석까지 모두 달아놓고 옮기기만 해야겠다. 문제의 난이도 자체는 크게 어렵지 않았으나 자바로 구현하려고 하니 버벅거려서 파이썬으로 풀었다. 이후에 자바로도 풀어봐야겠다. 문제를 푸는데 가장 중요한 것은 1. 기존의 string값을 배열로 변경, 2. 길이로 정렬, 3. 순차적으로 삽입 인 것 같다.

관련글 더보기

댓글 영역