코딩 문제 풀기 1

프로그래머스 문제를 풀어보자.

목차

1. 행렬의 덧셈
2. 홀짝 구하기
3. 정렬 k번째 수
4. 마라톤에 완주하지 못 한 선수
5. 전화번호 목록

1. 행렬의 덧셈

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 된다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

public class Level1 {

    public static void main(String[] args) {

        int[][] arr1 = { {1, 2}, {3, 3} };
        int[][] arr2 = { {2, 2}, {1, 3} };

        int[][] answer = solution(arr1, arr2);
        // answer = { {3, 4}, {4, 6} }
    }

    public static int[][] solution(int[][] arr1, int[][] arr2) {
        int[][] answer = new int[arr1.length][arr1[0].length];

        for(int i = 0; i < arr1.length; i++) {
            for(int j = 0; j < arr1[0].length; j++) {
                answer[i][j] = arr1[i][j] + arr2[i][j];
            }
        }

        return answer;
    }
}

• arr1.length는 행의 길이를 구한다.
• arr1[0].length는 열의 길이를 구한다.

2. 홀짝 구하기

숫자 num이 짝수이면 “Even”, 홀수이면 “Odd”를 반환하는 함수 solution을 완성해주세요.

public static String solution (int num) {
    String answer = "";

    if(num % 2 == 0) {
        answer = "Even";
    } else {
        answer = "Odd";
    }

    return answer;
}

3. 정렬 k번째 수

문제설명

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 합니다.

예를 들어 array가 [1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3이라면

array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]입니다. 1에서 나온 배열을 정렬하면 [2, 3, 5, 6]입니다. 2에서 나온 배열의 3번째 숫자는 5입니다. 배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한사항

• array의 길이는 1 이상 100 이하입니다.
• array의 각 원소는 1 이상 100 이하입니다.
• commands의 길이는 1 이상 50 이하입니다.
• commands의 각 원소는 길이가 3입니다.

입출력 예

• array [1, 5, 2, 6, 3, 7, 4]
• commands [[2, 5, 3], [4, 4, 1], [1, 7, 3]]
• return [5, 6, 3]

import java.util.Arrays;

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];

        for(int i = 0; i < commands.length; i++) {
            
            int[] tempArray = Arrays.copyOfRange(array, commands[i][0] - 1, commands[i][1]);
            Arrays.sort(tempArray);

            answer[i] = tempArray[commands[i][2] - 1];
        }

        return answer;
    }
}

위의 내용이 정답이고 처음에 Arrays.copyOfRange 메서드와 Arrays.sort 메서드를 사용했는데 실패했다. 이유는 import java.util.Arrays; 를 하지 않아서이다.

그래서 따로 배열을 자르는 메서드와 버블정렬 메서드를 구현했엇다. 근데 위의 메서드를 사용하면 필요가 없다.

public static int[] cutArray(int[] array, int startIndex, int endIndex) {
    int length = endIndex - startIndex + 1;
    int[] result = new int[length];

    int index = 0;
    for(int j = startIndex - 1; j < endIndex; j++) {
        result[index] = array[j];
        ++index;
    }

    return result;
}

public static int[] sortArray(int[] arr) {
    for(int i = 0; i < arr.length; i++) {
        for(int j = 0; j < arr.length -i -1; j++) {
            if(arr[j] > arr[j + 1]) {
                int temp = arr[j];
                arr[j] = arr[j + 1];
                arr[j + 1] = temp;
            }
        }
    }
    return arr;
}

4. 마라톤에 완주하지 못 한 선수

문제 설명

수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.

마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.

제한사항

마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다. completion의 길이는 participant의 길이보다 1 작습니다. 참가자의 이름은 1개 이상 20개 이하의 알파벳 소문자로 이루어져 있습니다. 참가자 중에는 동명이인이 있을 수 있습니다.

내가 작성한 코드

import java.util.Arrays;

class Solution {
    public String solution(String[] participant, String[] completion) {
        Arrays.sort(participant);
        Arrays.sort(completion);

        int i;

        for(i = 0; i < completion.length; i++) {
            if(!participant[i].equals(completion[i])) {
                return participant[i];
            }
        }

        return participant[i];
    }
}

다른 사람이 작성한 코드

public static String solution(String[] participant, String[] completion) {
    String answer = "";
    HashMap<String, Integer> hm = new HashMap<>();
    for (String player : participant)
        hm.put(player, hm.getOrDefault(player, 0) + 1);
    for (String player : completion)
        hm.put(player, hm.get(player) - 1);

    for (String key : hm.keySet()) {
        if (hm.get(key) != 0){
            answer = key;
        }
    }
    return answer;
}

• 해시맵을 사용하여 참가자 이름은 key로 중복 숫자를 value로 세팅한다.
• 완주자 루프를 돌면서 해당 완주한 참자가 이름을 찾아 value를 1씩 감소시킨다.
• 해시맵에서 value값이 0이 아닌 참자가의 key를 answer 변수에 넣고 리턴한다.
  
  

5. 전화번호 목록

문제 설명

전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다. 전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.

구조대 : 119 박준영 : 97 674 223 지영석 : 11 9552 4421 전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.

제한 사항

• phone_book의 길이는 1 이상 1,000,000 이하입니다.
• 각 전화번호의 길이는 1 이상 20 이하입니다.

입출력 예제

입출력 예 설명

입출력 예 #1 앞에서 설명한 예와 같습니다.

입출력 예 #2 한 번호가 다른 번호의 접두사인 경우가 없으므로, 답은 true입니다.

입출력 예 #3 첫 번째 전화번호, “12”가 두 번째 전화번호 “123”의 접두사입니다. 따라서 답은 false입니다.

내가 작성한 코드

public boolean solution(String[] phone_book) {
    boolean answer = true;
    Map<String, Integer> map = new HashMap<>();

    for(String s : phone_book) {
        map.put(s, map.getOrDefault(s, 0));
    }

    for(String k1 : map.keySet()) {
        for(String k2 : map.keySet()) {
            if(k2.startsWith(k1) && !k1.equals(k2)) {
                answer = false;
                break;
            }
        }
    }

    return answer;
}

문제 카테고리가 해시 라고 나와 있어서 Map을 사용했는데 효율성에서 문제가 발생하여 통과하지 못 했다.

다른 사람이 작성한 코드

public static boolean solution(String[] phone_book) {
    for(int i = 0; i < phone_book.length - 1; i++) {
        for(int j = i + 1; j <phone_book.length; j++) {
            if(phone_book[i].startsWith(phone_book[j])) return false;
            if(phone_book[j].startsWith(phone_book[i])) return false;
        }
    }
    return true;
}

contains가 아닌 startsWith를 써야 하는 이유는 contains는 prefix를 체크하지 않기 때문이다.