장난꾸러기

27 Apr 2023 in Algorithm on Java, Coding, Test

문제

설명

새 학기가 시작되었습니다. 철수는 새 짝꿍을 만나 너무 신이 났습니다.

철수네 반에는 N명의 학생들이 있습니다.

선생님은 반 학생들에게 반 번호를 정해 주기 위해 운동장에 반 학생들을 키가 가장 작은 학생부터 일렬로 키순으로 세웠습니다.

제일 앞에 가장 작은 학생부터 반 번호를 1번부터 N번까지 부여합니다. 철수는 짝꿍보다 키가 큽니다.

그런데 철수가 앞 번호를 받고 싶어 짝꿍과 자리를 바꿨습니다.

선생님은 이 사실을 모르고 학생들에게 서있는 순서대로 번호를 부여했습니다.

철수와 짝꿍이 자리를 바꾼 반 학생들의 일렬로 서있는 키 정보가 주어질 때 철수가 받은 번호와 철수 짝꿍이 받은 번호를

차례로 출력하는 프로그램을 작성하세요.

입력

첫 번째 줄에 자연수 N(5<=N<=100)이 주어진다.

두 번째 줄에 제일 앞에부터 일렬로 서있는 학생들의 키가 주어진다.

키(높이) 값 H는 (120<=H<=180)의 자연수 입니다.

출력

첫 번째 줄에 철수의 반 번호와 짝꿍의 반 번호를 차례로 출력합니다.

예시 입력 1

9
120 125 152 130 135 135 143 127 160

예시 출력 1

3 8

해결방법

• 정렬된 배열 중 일부 원소의 순서만 바뀐 상태에서 바뀐 순서의 원소 index를 알아내야 함.
• 복사본을 다시 정렬한 뒤 비교.

코드

import java.util.*;

class Main {
    public ArrayList<Integer> solution(int n, int[] arr) {
        // 배열로 선언
        ArrayList<Integer> answer = new ArrayList<>();
        int[] tmp = arr.clone();
        // 정렬하고 시작
        Arrays.sort(tmp);
        // 크기만큼 반복
        for (int i = 0; i < n; i++) {
            // 정렬된 상태에서 동일 위치 값이 다른 경우 정답 + 1
            if (arr[i] != tmp[i]) answer.add(i + 1);
        }
        return answer;
    }

    public static void main(String[] args) {
        Main.Main T = new Main.Main();
        Scanner kb = new Scanner(System.in);
        int n = kb.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) arr[i] = kb.nextInt();
        for (int x : T.solution(n, arr)) System.out.print(x + " ");
    }
}