문제

BOJ 1913: 달팽이 https://www.acmicpc.net/problem/1913

풀이

• (0, 0)의 위치에 N X N 값을 넣고 시작한다.
• 문제에 나와있는 조건대로 수를 1씩 줄여나가며 끝까지 넣어준다.
  • 반시계 방향으로 바깥 쪽에서부터 안 쪽으로 차례로 넣는다.
• 종료 조건 : 넣은 값이 1이 되면 종료한다.
• 완성한 배열을 출력할 때 구해야 하는 좌표를 구하여 함께 출력한다.

코드

import java.util.*;
import java.io.*;

public class Main {
    
	static int N, A;
	static int[][] arr;
	
	static int[] dx = {1, 0, -1, 0}; // 아래로, 오른쪽으로, 위로, 왼쪽으로 탐색
	static int[] dy = {0, 1, 0, -1};
	
    public static void main(String[] args) throws IOException {
    	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    	StringBuilder sb = new StringBuilder();
    	
    	N = Integer.parseInt(br.readLine()); // 달뱅이 한 변의 길이
    	A = Integer.parseInt(br.readLine()); // 찾아야 하는 수
    	
    	arr = new int[N][N]; // 출력할 N * N 배열
    	
    	int idx = 0;
    	
    	// 시작 좌표
    	int curX = 0; 
    	int curY = 0;
    	
    	arr[curX][curY] = N*N; // 시작할 숫자
    	
    	while(idx < 4) {
    		int nx = curX + dx[idx];
    		int ny = curY + dy[idx];
    		
    		// 다음 좌표가 범위 안이고 아직 수를 넣지 않은 위치라면
    		if(nx >= 0 && ny >= 0 && nx < N && ny < N && arr[nx][ny] == 0) {
    			// 이전 좌표에 있던 수 보다 1 작은 값을 새로운 좌표에 넣어준다.
    			arr[nx][ny] = arr[curX][curY] - 1;			
    			
    			// 종료 조건
    			if(arr[nx][ny] == 1) break; 
        		
    			// 다음 좌표를 찾기 위해 값을 갱신한다.
        		curX = nx;
        		curY = ny;
    		} 
    		// 다음 좌표가 범위를 벗어났거나 이미 수를 넣은 좌표라면 idx를 1 증가시켜준다.
    		else {
    			idx++;
    		}
    		
    		// idx 가 4 이상이 되면 다시 0으로 돌려줌
    		if(idx >= 4) {
    			idx = 0;
    		}
    			
    	}
    	
    	int ansX = 0, ansY = 0; // 찾아야 하는 수의 좌표 변수
    	// 정답 출력
    	for(int i=0; i<N; i++) {
    		for(int j=0; j<N; j++) {
    			// A 값의 좌표 구하기
    			if(arr[i][j] == A) {
    				ansX = i + 1;
    				ansY = j + 1;
    			}
    			sb.append(arr[i][j] + " ");
    		}
    		sb.append("\n");
    	}
    	sb.append(ansX + " " + ansY);

    	System.out.println(sb);
    }

}

정리

• 처음에 종료 조건을 넣어 주지 않아 시간 초과가 났다.