[코드트리/INTERMEDIATE LOW] 삼각형 컨베이어 벨트

📖 문제

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

 

💡  풀이 방식

• 구현, 시뮬레이션

1. 3 *N 크기의 2차원 배열을 입력받는다.

2. 각 줄마다 맨 오른쪽 값을 따로 저장해두고, 각 줄의 모든 값을 오른쪽으로 민다.

3. 각 줄의 맨 왼쪽의 빈 칸에 따로 저장해두었던 맨 오른쪽 값들을 맞는 자리에 넣는다.

4. 과정 2,3을 T번 반복한다.

4. 완성된 2차원 배열을 출력한다.

 

 

 

💥 유의사항

- 양쪽 끝에 있는 값 유실되지 않도록 따로 저장해 처리해야 한다.

- 가장 끝 칸부터 앞으로 오며 순서대로 채워줘야 정보 유실을 막는다.

 

 

🔺 코드

import java.util.*;
import java.io.*;
 
public class Main {
    static int N, T;
    static int[][] arr;
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
 
        N = Integer.parseInt(st.nextToken());
        T = Integer.parseInt(st.nextToken());
 
        arr = new int[3][N];
        
        for(int i = 0 ; i < 3 ; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            for(int j = 0 ; j < N ; j++) {
                arr[i][j] = Integer.parseInt(st.nextToken());
            }            
        }
 
        while(T --> 0) {
            rotate();
        }
 
        print();
    }
 
    // 회전 메소드
    private static void rotate() {
        int first = arr[0][N - 1];    // 첫 번째 줄의 가장 오른쪽에 있는 숫자
        for(int i = N - 1 ; i >= 1 ; i--) {    // 숫자들 오른쪽으로 밀기
            arr[0][i] = arr[0][i - 1];
        }
        
        int second = arr[1][N - 1];    // 두 번째 줄의 가장 오른쪽에 있는 숫자
        for(int i = N - 1 ; i >= 1 ; i--) {    // 숫자들 오른쪽으로 밀기
            arr[1][i] = arr[1][i - 1];
        }
 
        int third = arr[2][N - 1];    // 세 번째 줄의 가장 오른쪽에 있는 숫자
        for(int i = N - 1 ; i >= 1 ; i--) {    // 숫자들 오른쪽으로 밀기
            arr[2][i] = arr[2][i - 1];
        }
 
        // 각 줄의 맨 왼쪽 숫자 채우기 (따로 저장했던 각 줄의 가장 오른쪽에 있는 숫자들을 각각의 왼쪽 위치에 넣기)
        arr[0][0] = third;
        arr[1][0] = first;
        arr[2][0] = second;
    }
 
    // 결과 출력 메소드
    private static void print() {
        StringBuilder sb = new StringBuilder();
        for(int[] i : arr) {
            for(int j : i) 
                sb.append(j + " ");
            sb.append("\n");
        }
 
        System.out.println(sb);
    }
}

 

 

 

➕ 다른 풀이 방식

각 줄을 1차원 배열로 만들어 총 3개의 1차원 배열을 만들어 푸시기도 하셨다.

(146ms, 9MB)

import java.util.*;
import java.io.*;
 
public class Main {
    static int n, t;
 
    static int [] arr; // 왼쪽 위
    static int [] brr; // 오른쪽 위
    static int [] crr; // 밑 변
 
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
 
        n = Integer.parseInt(stk.nextToken());
        t = Integer.parseInt(stk.nextToken());
 
        arr = new int[n];
        brr = new int[n];
        crr = new int[n];
 
        st = new StringTokenizer(br.readLine());
        for(int i = 0 ; i < n ; i++){
            arr[i] = Integer.parseInt(stk.nextToken());
        }
        
        st = new StringTokenizer(br.readLine());
        for(int i = 0 ; i < n ; i++){
            brr[i] = Integer.parseInt(stk.nextToken());
        }
        
        st = new StringTokenizer(br.readLine());
        for(int i = 0 ; i < n ; i++){
            crr[i] = Integer.parseInt(stk.nextToken());
        }
 
        while(t-- > 0){
            int arrT = arr[n-1];
            int brrT = brr[n-1];
            int crrT = crr[n-1];
            
            for(int i = n-1;i>0;i--){
                arr[i] = arr[i-1];
                brr[i] = brr[i-1];
                crr[i] = crr[i-1];
            }
 
            arr[0] = crrT;
            brr[0] = arrT;
            crr[0] = brrT;
        }
 
        StringBuilder sb = new StringBuilder();
        for(int a : arr){
            sb.append(a).append(" ");
        }
        sb.append("\n");
 
        for(int a : brr){
            sb.append(a).append(" ");
        }
        sb.append("\n");
 
        for(int a : crr){
            sb.append(a).append(" ");
        }
        System.out.print(sb);
 
 
    }
}

 

 

---

💦 어려웠던 점

초반에 약간 모양에 쫄았으나 쫄 필요 없음!

 

 

 

1회독	2회독	3회독	4회독	5회독
V