[백준] 10813 : 공바꾸기 (setAll() 메소드) - JAVA

백준 단계별 문제풀이 4단계 (1차원 배열)

https://www.acmicpc.net/problem/10813

10813번: 공 바꾸기

 

# 문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 바구니에는 공이 1개씩 들어있고, 처음에는 바구니에 적혀있는 번호와 같은 번호가 적힌 공이 들어있다.

도현이는 앞으로 M번 공을 바꾸려고 한다. 도현이는 공을 바꿀 바구니 2개를 선택하고, 두 바구니에 들어있는 공을 서로 교환한다.

공을 어떻게 바꿀지가 주어졌을 때, M번 공을 바꾼 이후에 각 바구니에 어떤 공이 들어있는지 구하는 프로그램을 작성하시오.

 

# 예제

입력 -

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에 걸쳐서 공을 교환할 방법이 주어진다. 각 방법은 두 정수 i j로 이루어져 있으며, i번 바구니와 j번 바구니에 들어있는 공을 교환한다는 뜻이다.

5 4
1 2
3 4
1 4
2 2

 

출력

3 1 4 2 5

 

# 필요개념

swap메소드를 간단하게 구현해두고, 바꿀 공의 숫자를 받아 숫자 변경을 했다.
배열을 [1,2,3,4,5]로 세팅할 필요가 있었는데, 이때는 Arrays.setAll() 메소드를 사용했다.

setAll 메소드에는 첫 인수로 배열을 넣어주고, 두번째 인수로 람다 함수식을 넣어 간단히 작성할 수 있다!

setAll 메소드는 람다식으로 넣은 함수를 이용해 값을 할당해주는 메소드라고 생각하면 된다. 예를 들어 람다식으로 

i -> i + 1 을 사용했다고 가정해보자. i는 인덱스라고 생각하면 편하다. 따라서 해당 람다식은 인덱스보다 1씩 큰 1, 2, 3... 을 결과로 내놓는 역할을 한다.

 

이와 비슷한 메소드로 Arrays.fill() 이 있는데, fill은 지정한 정수로 모두 채워주는 역할을 한다. fill은 보통 배열을 초기화 할 때 사용한다고 생각하면 된다.

 

결론적으로는 setAll을 사용함으로써 반복문을 이용해 배열을 채우지 않아도 되었다!

 

#  Code

import java.util.*;

public class Main {
    public static void swap(int a, int b, int[] arry) {
        int tmp;
        tmp = arry[a];
        arry[a] = arry[b];
        arry[b] = tmp;
    }
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int size = sc.nextInt();
        int n = sc.nextInt();
        int[] arr = new int[size];
        Arrays.setAll(arr, i -> i + 1);

        for (int i = 0 ; i < n ; i++) {
            int x = sc.nextInt();
            int y = sc.nextInt();
            swap(x - 1, y - 1, arr);

        }
        Arrays.stream(arr).forEach(System.out::println);
    }
}

 

#  결과