👩‍💻 BOJ_1913_달팽이

💬 열과 행의 증가/감소에 따른 4가지 경우를 생각해 문제를 풀었습니다. 시간초과의 문제는 Scanner -> BuffererdReader/Writer 로 변경하여 해결하였습니다.

📄 문제

홀수인 자연수 N에 대해 1부터 N*N까지의 자연수를 달팽이 모양으로 N×N의 표현하고, K의 값이 존재하는 위치를 출력하시오.

💡 풀이 방법

• 달팽이 행렬을 채울 수 있는 4가지 경우를 구현하였습니다.
  1) 행 증가, 열 유지 : i++ / j
  2) 행 유지, 열 증가 : i / j++
  3) 행 감소, 열 유지 : i-- / j
  4) 행 유지, 열 감소 : i / j--

• 채운 행렬을 출력합니다.

  이 때, System.out.print(); 를 사용하니 시간 초과가 나왔습니다. 그래서 출력 방법을 변경하였습니다.

🔥 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
//import java.util.Scanner;

public class 달팽이 {

	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		//Scanner sc = new Scanner(System.in);
		
		//int N = sc.nextInt(); // 달팽이 사이즈
		//int search = sc.nextInt(); // 찾을 값
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int N = Integer.parseInt(br.readLine()); // 달팽이 사이즈
		int search = Integer.parseInt(br.readLine()); // 찾을 값 
		
		int start = N*N; // 최초 값
		
		int snail[][] = new int[N+1][N+1];
		int visit[][] = new int[N+1][N+1];
		int loc_i = 1;
		int tmp_i = 1; 
		int loc_j = 1;
		int tmp_j = 1;
		
		
		while(true) {
			// 행 증가, 열 유지 : i++, j 
			for(int i = loc_i; i < N+1; i++) {
				if(visit[i][loc_j] == 0) {
					snail[i][loc_j] = start;
					start--;
					visit[i][loc_j] = 1;
					tmp_i = i;
				}
			}
			loc_i = tmp_i;
			
			// 행 유지, 열 증가 : i, j++
			for(int j = loc_j; j < N+1; j++) {
				if(visit[loc_i][j] == 0) {
					snail[loc_i][j] = start;
					start--;
					visit[loc_i][j] = 1;
					tmp_j = j;
				}
			}
			loc_j = tmp_j;
			
			// 행 감소, 열 유지 : i--, j
			for(int i = loc_i; i >= 1; i--) {
				if(visit[i][loc_j] == 0) {
					snail[i][loc_j] = start;
					start--;
					visit[i][loc_j] = 1;
					tmp_i = i;
				}
			}
			loc_i = tmp_i;
			
			// 행 유지, 열 감소 : i, j--
			for(int j = loc_j; j >= 1; j--) {
				if(visit[loc_i][j] == 0) {
					snail[loc_i][j] = start;
					start--;
					visit[loc_i][j] = 1;
					tmp_j = j;
				}
			}
			loc_j = tmp_j;
			
			if(start == 0) {
				break;
			}
		}
		
		// 정답 추출
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));   //할당된 버퍼에 값 넣어주기
		
		for(int i = 1; i < N+1; i++) {
			for(int j = 1; j < N+1; j++) {
				bw.write(snail[i][j] + " ");
				//System.out.print(snail[i][j]+" ");
				if(snail[i][j] == search) {
					loc_i = i;
					loc_j = j;
				}
			}
			bw.write("\n");
			//System.out.println();
		}
		bw.write(loc_i + " " + loc_j);
		//System.out.print(loc_i + " " + loc_j);
		bw.flush();   // 남아있는 데이터를 모두 출력시킴
		bw.close();   // 스트림을 닫음
		
	}

}

⭐️ 오늘의 학습

BufferedWriter 사용법

1. BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
2. bw.write(출력 문자열);

• BufferedWriter 의 경우 반드시 flush() / close() 를 호출해 주어 버퍼를 정리해야한다.
• System.out.println(); 과 같은 개행 기능을 사용하려면 bw.write("\n"); 을 써준다.