[Java] 장난꾸러기

문제

풀이

철수가 받은 번호와 짝궁이 받은 번호를 찾는 문제이다. 입력으로 주어진 배열을 임시 배열에 복사를 한다. 그리고 원소별로 비교하면 철수와 짝궁의 번호를 알 수 있다.

코드

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Scanner;

public class Main {
    public ArrayList<Integer> solution(int n, int[] arr) {
        ArrayList<Integer> answer = new ArrayList<>();
        int[] temp = arr.clone();
        Arrays.sort(temp);  // 정렬

        for (int i=0; i<n; i++) {
            if (arr[i] != temp[i])  // 정렬 후 비교
                answer.add(i+1);
        }
        return answer;
    }


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