백준 16926번 : 배열 돌리기 1

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

16926번: 배열 돌리기 1

 

 

 

 

 

풀면서 화나는 문제이다..

회전 수만큼 for문 을 돌려,

안에서 ← , ↓, →, ↑ 네 방향으로 나눠 배열의 값을 방향대로 옮기기로 했다.

인덱스 때문에 애를 먹었다. 프린트문을 찍어가며 결과값을 확인하며 조금씩 고친 끝에 정답을 낼 수 있었다. 

 

 

 

 

 

실패 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
	static int N, M;
	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());
		M = Integer.parseInt(st.nextToken());
		int R = Integer.parseInt(st.nextToken());
		
		arr = new int[N][M];

		for(int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine(), " ");
			
			for(int j = 0; j < M; j++) {
				arr[i][j] = Integer.parseInt(st.nextToken());
			}
		}
		
		//구현
		for(int r = 0; r < R; r++) {
			rotate();
		}
		
		
		//출력
		for(int i = 0; i < N; i++) {
			for(int j = 0; j < M; j++) {
				System.out.print(arr[i][j] + " ");
			}
			System.out.println();
		}
		
	}

	public static void rotate() {
		for(int i = 0; i < N / 2; i++) { //회전 묶음만큼 반복
			int temp1 = arr[i][M - 1 - i];
			int temp2 = 0;
			
			for(int j = M - 2 - i; j >= i; j--) { //← 방향
				temp2 = arr[i][j];
				arr[i][j] = temp1;
				temp1 = temp2;
			}
			
			for(int j = i + 1; j < N - i; j++) { //↓ 방향
				temp2 = arr[j][i];
				arr[j][i] = temp1;
				temp1 = temp2;
			}
			
			for(int j = i + 1; j < M - i; j++) { //→ 방향
				temp2 = arr[N - 1 - i][j];
				arr[N - 1 - i][j] = temp1;
				temp1 = temp2;
			}
			
			for(int j = N - 2 - i; j >= i; j--) { //↑ 방향
				temp2 = arr[j][M - 1 - i];
				arr[j][M - 1 - i] = temp1;
				temp1 = temp2;
			}
		}
		
	}

}

 

 

회전수만큼 반복해주는 것을 N/2로 했는데, 무조건 N이 아니라 M과 N 중에 작은 값을 써야 한다. 

 

 

 

 

 

성공 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
	static int N, M;
	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());
		M = Integer.parseInt(st.nextToken());
		int R = Integer.parseInt(st.nextToken());
		
		arr = new int[N][M];

		for(int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine(), " ");
			
			for(int j = 0; j < M; j++) {
				arr[i][j] = Integer.parseInt(st.nextToken());
			}
		}
		
		//구현
		for(int r = 0; r < R; r++) {
			rotate();
		}
		
		
		//출력
		StringBuilder sb = new StringBuilder();
		for(int i = 0; i < N; i++) {
			for(int j = 0; j < M; j++) {
				sb.append(arr[i][j] + " ");
			}
			sb.append("\n");
		}
		
		System.out.println(sb);
		
	}

	public static void rotate() {
        int d = Math.min(N, M);
        
		for(int i = 0; i < d / 2; i++) { //회전 묶음만큼 반복
			int temp1 = arr[i][M - 1 - i];
			int temp2 = 0;
			
			for(int j = M - 2 - i; j >= i; j--) { //← 방향
				temp2 = arr[i][j];
				arr[i][j] = temp1;
				temp1 = temp2;
			}
			
			for(int j = i + 1; j < N - i; j++) { //↓ 방향
				temp2 = arr[j][i];
				arr[j][i] = temp1;
				temp1 = temp2;
			}
			
			for(int j = i + 1; j < M - i; j++) { //→ 방향
				temp2 = arr[N - 1 - i][j];
				arr[N - 1 - i][j] = temp1;
				temp1 = temp2;
			}
			
			for(int j = N - 2 - i; j >= i; j--) { //↑ 방향
				temp2 = arr[j][M - 1 - i];
				arr[j][M - 1 - i] = temp1;
				temp1 = temp2;
			}
		}
		
	}

}

 

 

 

 

 

 

결과

 

아래는 출력을 System.out.println()으로 한 것이고, 위에 것은 StringBuilder를 이용해 출력한 것이다.

입력값이 크므로 StringBuilder를 사용하는 것이 시간을 줄이는 데 유용하다.