99클럽 코테 스터디 20일차 TIL + 프로그래머스 모의고사, 백준 소트

 

 

문제

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.

1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한 조건

• 시험은 최대 10,000 문제로 구성되어있습니다..
• 문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
• 가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.
  
  

import java.util.ArrayList;

public class Solution {
    public int[] solution(int[] answers){


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

        int[] count = new int[3];

        for (int i=0;i<answers.length;i++){
            if(arr1[i%5] == answers[i]){
                count[0] +=1;
            }
            if(arr2[i%8] == answers[i]){
                count[1] +=1;
            }
            if(arr3[i%10] == answers[i]){
                count[2] +=1;
            }
        }

        int max = count[0];
        for(int i=0;i<3;i++){
            if(max<=count[i]){
                max = count[i];
            }
        }

        ArrayList<Integer> resultList = new ArrayList<>();
        for(int i=0;i<3;i++){
            if(max== count[i]){
                resultList.add(i+1);
            }
        }

        int[] result = new int[resultList.size()];
        for(int i=0;i<resultList.size();i++){
            result[i] = resultList.get(i);
        }

        return result;
    }
}

 

---

간단한 완전탐색 문제였다.

아래는 Math나 stream을 활용하여 좀 더 간략하게 정리한 코드

import java.util.ArrayList;
import java.util.List;

public class Solution {
    public int[] solution(int[] answers) {
        int[] arr1 = {1, 2, 3, 4, 5};
        int[] arr2 = {2, 1, 2, 3, 2, 4, 2, 5};
        int[] arr3 = {3, 3, 1, 1, 2, 2, 4, 4, 5, 5};

        int[] count = new int[3];

        for (int i = 0; i < answers.length; i++) {
            if (arr1[i % arr1.length] == answers[i]) count[0]++;
            if (arr2[i % arr2.length] == answers[i]) count[1]++;
            if (arr3[i % arr3.length] == answers[i]) count[2]++;
        }

        int max = Math.max(count[0], Math.max(count[1], count[2]));
 
        List<Integer> resultList = new ArrayList<>();
        for (int i = 0; i < count.length; i++) {
            if (count[i] == max) {
                resultList.add(i + 1);
            }
        }

        return resultList.stream().mapToInt(i -> i).toArray();
    }
}

 

 

---

 

 

문제

크기가 N인 배열 A가 있다. 배열에 있는 모든 수는 서로 다르다. 이 배열을 소트할 때, 연속된 두 개의 원소만 교환할 수 있다. 그리고, 교환은 많아봐야 S번 할 수 있다. 이때, 소트한 결과가 사전순으로 가장 뒷서는 것을 출력한다.

 

입력

첫째 줄에 N이 주어진다. N은 50보다 작거나 같은 자연수이다. 둘째 줄에는 각 원소가 차례대로 주어진다. 이 값은 1000000보다 작거나 같은 자연수이다. 마지막 줄에는 S가 주어진다. S는 1000000보다 작거나 같은 음이 아닌 정수이다.

 

출력

첫째 줄에 문제의 정답을 출력한다.

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int n = Integer.parseInt(br.readLine());

        int[] arr = new int[n];

        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }

        int s = Integer.parseInt(br.readLine());

        for (int i = 0; i < n && s > 0; i++) {

            //현재 위치에서 최대값 찾기 (s 범위 내에서)
            int maxIndex = i;
            for (int j = i + 1; j < n && j <= i + s; j++) {
                if (arr[j] > arr[maxIndex]) {
                    maxIndex = j;
                }
            }

            if (maxIndex == i) {
                continue;
            }

            for (int j = maxIndex; j > i; j--) {
                int temp = arr[j];
                arr[j] = arr[j - 1];
                arr[j - 1] = temp;
                s--;
            }

            if (s <= 0) {
                break;
            }
        }

        for (int index : arr) {
            System.out.print(index + " ");
        }

    }

}