[Java] 백준 16935번 : 배열 돌리기 3

코딩테스트/구현(Implementation)

[Java] 백준 16935번 : 배열 돌리기 3

sujin7837 2022. 2. 11. 22:15

문제

크기가 N×M인 배열이 있을 때, 배열에 연산을 R번 적용하려고 한다. 연산은 총 6가지가 있다.

1번 연산은 배열을 상하 반전시키는 연산이다.

1 6 2 9 8 4 → 4 2 9 3 1 8
7 2 6 9 8 2 → 9 2 3 6 1 5
1 8 3 4 2 9 → 7 4 6 2 3 1
7 4 6 2 3 1 → 1 8 3 4 2 9
9 2 3 6 1 5 → 7 2 6 9 8 2
4 2 9 3 1 8 → 1 6 2 9 8 4
   <배열>       <연산 결과>

2번 연산은 배열을 좌우 반전시키는 연산이다.

1 6 2 9 8 4 → 4 8 9 2 6 1
7 2 6 9 8 2 → 2 8 9 6 2 7
1 8 3 4 2 9 → 9 2 4 3 8 1
7 4 6 2 3 1 → 1 3 2 6 4 7
9 2 3 6 1 5 → 5 1 6 3 2 9
4 2 9 3 1 8 → 8 1 3 9 2 4
   <배열>       <연산 결과>

3번 연산은 오른쪽으로 90도 회전시키는 연산이다.

1 6 2 9 8 4 → 4 9 7 1 7 1
7 2 6 9 8 2 → 2 2 4 8 2 6
1 8 3 4 2 9 → 9 3 6 3 6 2
7 4 6 2 3 1 → 3 6 2 4 9 9
9 2 3 6 1 5 → 1 1 3 2 8 8
4 2 9 3 1 8 → 8 5 1 9 2 4
   <배열>       <연산 결과>

4번 연산은 왼쪽으로 90도 회전시키는 연산이다.

1 6 2 9 8 4 → 4 2 9 1 5 8
7 2 6 9 8 2 → 8 8 2 3 1 1
1 8 3 4 2 9 → 9 9 4 2 6 3
7 4 6 2 3 1 → 2 6 3 6 3 9
9 2 3 6 1 5 → 6 2 8 4 2 2
4 2 9 3 1 8 → 1 7 1 7 9 4
   <배열>       <연산 결과>

5, 6번 연산을 수행하려면 배열을 크기가 N/2×M/2인 4개의 부분 배열로 나눠야 한다. 아래 그림은 크기가 6×8인 배열을 4개의 그룹으로 나눈 것이고, 1부터 4까지의 수로 나타냈다.

1 1 1 1 2 2 2 2
1 1 1 1 2 2 2 2
1 1 1 1 2 2 2 2
4 4 4 4 3 3 3 3
4 4 4 4 3 3 3 3
4 4 4 4 3 3 3 3

5번 연산은 1번 그룹의 부분 배열을 2번 그룹 위치로, 2번을 3번으로, 3번을 4번으로, 4번을 1번으로 이동시키는 연산이다.

3 2 6 3 1 2 9 7 → 2 1 3 8 3 2 6 3
9 7 8 2 1 4 5 3 → 1 3 2 8 9 7 8 2
5 9 2 1 9 6 1 8 → 4 5 1 9 5 9 2 1
2 1 3 8 6 3 9 2 → 6 3 9 2 1 2 9 7
1 3 2 8 7 9 2 1 → 7 9 2 1 1 4 5 3
4 5 1 9 8 2 1 3 → 8 2 1 3 9 6 1 8
     <배열>            <연산 결과>

6번 연산은 1번 그룹의 부분 배열을 4번 그룹 위치로, 4번을 3번으로, 3번을 2번으로, 2번을 1번으로 이동시키는 연산이다.

3 2 6 3 1 2 9 7 → 1 2 9 7 6 3 9 2
9 7 8 2 1 4 5 3 → 1 4 5 3 7 9 2 1
5 9 2 1 9 6 1 8 → 9 6 1 8 8 2 1 3
2 1 3 8 6 3 9 2 → 3 2 6 3 2 1 3 8
1 3 2 8 7 9 2 1 → 9 7 8 2 1 3 2 8
4 5 1 9 8 2 1 3 → 5 9 2 1 4 5 1 9
     <배열>            <연산 결과>

입력

첫째 줄에 배열의 크기 N, M과 수행해야 하는 연산의 수 R이 주어진다.

둘째 줄부터 N개의 줄에 배열 A의 원소 Aij가 주어진다.

마지막 줄에는 수행해야 하는 연산이 주어진다. 연산은 공백으로 구분되어져 있고, 문제에서 설명한 연산 번호이며, 순서대로 적용시켜야 한다.

출력

입력으로 주어진 배열에 R개의 연산을 순서대로 수행한 결과를 출력한다.

제한

2 ≤ N, M ≤ 100
1 ≤ R ≤ 1,000
N, M은 짝수
1 ≤ Aij ≤ 108

예제 입력 1

6 8 1
3 2 6 3 1 2 9 7
9 7 8 2 1 4 5 3
5 9 2 1 9 6 1 8
2 1 3 8 6 3 9 2
1 3 2 8 7 9 2 1
4 5 1 9 8 2 1 3
1

예제 출력 1

4 5 1 9 8 2 1 3
1 3 2 8 7 9 2 1
2 1 3 8 6 3 9 2
5 9 2 1 9 6 1 8
9 7 8 2 1 4 5 3
3 2 6 3 1 2 9 7

예제 입력 2

6 8 1
3 2 6 3 1 2 9 7
9 7 8 2 1 4 5 3
5 9 2 1 9 6 1 8
2 1 3 8 6 3 9 2
1 3 2 8 7 9 2 1
4 5 1 9 8 2 1 3
2

예제 출력 2

7 9 2 1 3 6 2 3
3 5 4 1 2 8 7 9
8 1 6 9 1 2 9 5
2 9 3 6 8 3 1 2
1 2 9 7 8 2 3 1
3 1 2 8 9 1 5 4

예제 입력 3

6 8 1
3 2 6 3 1 2 9 7
9 7 8 2 1 4 5 3
5 9 2 1 9 6 1 8
2 1 3 8 6 3 9 2
1 3 2 8 7 9 2 1
4 5 1 9 8 2 1 3
3

예제 출력 3

4 1 2 5 9 3
5 3 1 9 7 2
1 2 3 2 8 6
9 8 8 1 2 3
8 7 6 9 1 1
2 9 3 6 4 2
1 2 9 1 5 9
3 1 2 8 3 7

예제 입력 4

6 8 1
3 2 6 3 1 2 9 7
9 7 8 2 1 4 5 3
5 9 2 1 9 6 1 8
2 1 3 8 6 3 9 2
1 3 2 8 7 9 2 1
4 5 1 9 8 2 1 3
4

예제 출력 4

7 3 8 2 1 3
9 5 1 9 2 1
2 4 6 3 9 2
1 1 9 6 7 8
3 2 1 8 8 9
6 8 2 3 2 1
2 7 9 1 3 5
3 9 5 2 1 4

예제 입력 5

6 8 1
3 2 6 3 1 2 9 7
9 7 8 2 1 4 5 3
5 9 2 1 9 6 1 8
2 1 3 8 6 3 9 2
1 3 2 8 7 9 2 1
4 5 1 9 8 2 1 3
5

예제 출력 5

2 1 3 8 3 2 6 3
1 3 2 8 9 7 8 2
4 5 1 9 5 9 2 1
6 3 9 2 1 2 9 7
7 9 2 1 1 4 5 3
8 2 1 3 9 6 1 8

예제 입력 6

6 8 1
3 2 6 3 1 2 9 7
9 7 8 2 1 4 5 3
5 9 2 1 9 6 1 8
2 1 3 8 6 3 9 2
1 3 2 8 7 9 2 1
4 5 1 9 8 2 1 3
6

예제 출력 6

1 2 9 7 6 3 9 2
1 4 5 3 7 9 2 1
9 6 1 8 8 2 1 3
3 2 6 3 2 1 3 8
9 7 8 2 1 3 2 8
5 9 2 1 4 5 1 9

예제 입력 7

6 8 6
3 2 6 3 1 2 9 7
9 7 8 2 1 4 5 3
5 9 2 1 9 6 1 8
2 1 3 8 6 3 9 2
1 3 2 8 7 9 2 1
4 5 1 9 8 2 1 3
1 2 3 4 5 6

예제 출력 7

3 1 2 8 9 1 5 4
1 2 9 7 8 2 3 1
2 9 3 6 8 3 1 2
8 1 6 9 1 2 9 5
3 5 4 1 2 8 7 9
7 9 2 1 3 6 2 3

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.StringTokenizer;

public class Main {

	private static int N, M, R, bigSize;
	private static int[][] arr;
	private static int[][] arrTmp;
	private static List<Integer> NUM = new ArrayList<Integer>();
	private static StringTokenizer st;
	
	public static void main(String[] args) throws IOException {

		BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));
		st=new StringTokenizer(bf.readLine());
		N=Integer.parseInt(st.nextToken());
		M=Integer.parseInt(st.nextToken());
		R=Integer.parseInt(st.nextToken());
		bigSize=Math.max(N, M);
		arr=new int[bigSize][bigSize];
		
		for(int r=0;r<N;r++) {
			st=new StringTokenizer(bf.readLine());
			for(int c=0;c<M;c++) {
				arr[r][c]=Integer.parseInt(st.nextToken());
			}
		}
		st=new StringTokenizer(bf.readLine());
		while(st.hasMoreTokens()) {
			NUM.add(Integer.parseInt(st.nextToken()));
		}
		
		for(int r:NUM) {
			choice(r);
			
		}
		
		for(int []r:arr) {
			for(int c:r) {
				if(c==0) continue;
				System.out.print(c+" ");
			}
			System.out.println();
		}
	}

	public static void choice(int n) {
		switch(n) {
		case 1:
			op1();
			break;
		case 2:
			op2();
			break;
		case 3:
			op3();
			break;
		case 4:
			op4();
			break;
		case 5:
			op5();
			break;
		case 6:
			op6();
			break;
		default:
			break;
		}
		
				
	}
	// 상하 반전
	public static void op1() {
		for(int c=0;c<M;c++) {
			for(int r=0;r<N/2;r++) {
				int tmp=arr[r][c];
				arr[r][c]=arr[N-1-r][c];
				arr[N-1-r][c]=tmp;
			}
		}
	}
	
	// 좌우 반전
	public static void op2() {
		for(int r=0;r<N;r++) {
			for(int c=0;c<M/2;c++) {
				int tmp=arr[r][c];
				arr[r][c]=arr[r][M-1-c];
				arr[r][M-1-c]=tmp;
			}
		}
	}
	
	// 오른쪽으로 90도
	public static void op3() {
		int tmp=N;
		N=M;
		M=tmp;
		arrTmp=new int[N][M];
		for(int r=0;r<N;r++) {
			for(int c=0;c<M;c++) arrTmp[r][c]=arr[M-c-1][r];
		}
		arr=new int[bigSize][bigSize];
		for(int r=0;r<N;r++) {
			for(int c=0;c<M;c++) arr[r][c]=arrTmp[r][c];
		}
	}
	
	// 왼쪽으로 90도
	public static void op4() {
		int tmp=N;
		N=M;
		M=tmp;
		arrTmp=new int[N][M];
		for(int r=0;r<N;r++) {
			for(int c=0;c<M;c++) arrTmp[r][c]=arr[c][N-r-1];
		}
		arr=new int[bigSize][bigSize];
		for(int r=0;r<N;r++) {
			for(int c=0;c<M;c++) arr[r][c]=arrTmp[r][c];
		}
	}
	
	// 그룹 연산
	// 그룹 시계 방향으로
	public static void op5() {
		arrTmp=new int[N][M];
		int [] dx = {0, N/2, 0, -N/2};
		int [] dy = {M/2, 0, -M/2, 0};
		int [] ox = {0, 0, N/2, N/2};
		int [] oy = {0, M/2, M/2, 0};
		int idx=0;
		
		for(int i=0;i<4;i++) {
			for(int r=ox[i];r<ox[i]+N/2;r++) {
				for(int c=oy[i];c<oy[i]+M/2;c++) {
					arrTmp[r+dx[i]][c+dy[i]]=arr[r][c];
				}
			}
		}
		arr=new int[bigSize][bigSize];
		for(int r=0;r<N;r++) {
			for(int c=0;c<M;c++) arr[r][c]=arrTmp[r][c];
		}
	}
	
	// 그룹 반시계 방향으로
	public static void op6() {
		arrTmp=new int[N][M];
		int [] dx = {N/2, 0, -N/2, 0};
		int [] dy = {0, M/2, 0, -M/2};
		int [] ox = {0, N/2, N/2, 0};
		int [] oy = {0, 0, M/2, M/2};
		int idx=0;
		
		for(int i=0;i<4;i++) {
			for(int r=ox[i];r<ox[i]+N/2;r++) {
				for(int c=oy[i];c<oy[i]+M/2;c++) {
					arrTmp[r+dx[i]][c+dy[i]]=arr[r][c];
				}
			}
		}
		arr=new int[bigSize][bigSize];
		for(int r=0;r<N;r++) {
			for(int c=0;c<M;c++) arr[r][c]=arrTmp[r][c];
		}
	}
}

'코딩테스트 > 구현(Implementation)' 카테고리의 다른 글

[Java] 백준 2635번 : 수 이어가기 (0)	2022.02.24
[Java] 백준 15686번 : 치킨 배달 (0)	2022.02.23
[Java] 백준 2563번 : 색종이 (0)	2022.02.10
[Java] 백준 16927번 : 배열 돌리기 2 (0)	2022.02.09
[Java] 백준 2456 : 나는 학급회장이다 (0)	2022.01.26

현재글[Java] 백준 16935번 : 배열 돌리기 3

sujin's 개발 로그