[알고리즘]Sorting & Searching - 장난꾸러기

6. 장난꾸러기
철수와 짝꿍이 자리를 바꾼 반 학생들의 일렬로 서있는 키 정보가 주어질 때 철수가 받은 번호와 철수 짝꿍이 받은 번호를 차례로 출력하는 프로그램

입력) 첫 번째 줄에 자연수 N(5<=N<=100)이 주어집니다.
두 번째 줄에 제일 앞에부터 일렬로 서있는 학생들의 키가 주어집니다.
출력) 첫 번째 줄에 철수의 반 번호와 짝꿍의 반 번호를 차례로 출력합니다.

 

 

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

public class Main {
	
	public static 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) {
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int[] arr = new int[n]; // 학생들의 키
		for (int i=0; i<n; i++) {
			arr[i] = scanner.nextInt();
		}
		for (int x : solution(n,arr)) {
			System.out.print(x + " ");
		}
	}

}

---

결과

입력
9
120 125 152 130 135 135 143 127 160

출력
3 8