[SWEA] 1954 달팽이 숫자 - Java

삼성 기출에 종종 나온 적이 있는 구현 문제.  

(0, 0)에서 시작 

 

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=2&problemLevel=3&problemLevel=4&contestProbId=AV5PobmqAPoDFAUq&categoryId=AV5PobmqAPoDFAUq&categoryType=CODE&problemTitle=&orderBy=SUBMIT_COUNT&selectCodeLang=JAVA&select-1=4&pageSize=10&pageIndex=1

SW Expert Academy

3*3 달팽이

4*4 달팽이

---

 📍 고려해야할 점

• 다음 칸에 갈 수 있는지 확인하며 방향을 바꿔야한다.  
  1. 칸의 범위를 넘어갈 때
  2. 다음 칸이 채워졌을 때
• 위의 두 가지 경우 중 하나라도 만족한다면 방향을 바꾼다. (우 ➔ 하 ➔ 좌 ➔ 상 순서대로)

 

🕹️ 풀이과정

1. 우 ➔ 하 ➔ 좌 ➔ 상 순서대로 달팽이 모양을 그리며 이동하므로 dx, dy 배열을 설정한다. 
2. 방향을 바꾸는 조건을 if 문을 활용하여 설정한다. 
3. 방향 바꾸는 것을 모듈러 연산을 활용하여 간단하게 나타낸다. (0 ➔ 1 ➔ 2 ➔ 3 ➔ 0) dir = (dir+1)%4
4. 결과를 출력한다. 

 

👩‍💻 전체 코드 

 

public class SWEA_1954_달팽이숫자 {
public static void main(String[] args) throws Exception{
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;
		
		int[] dx = {0, 1, 0, -1};
		int[] dy = {1, 0, -1, 0};
		
		int T = Integer.parseInt(br.readLine());
		
		for(int t=1; t<=T; t++) {
			
			st = new StringTokenizer(br.readLine());
			
			int N = Integer.parseInt(st.nextToken());
			
			int[][] arr=new int[N][N];
			
			int dir=0;
			int x=0;
			int y=0;
		
			for(int i=1; i<=N*N; i++) {
				arr[x][y] = i;
				
				int nx = x + dx[dir];
				int ny = y + dy[dir];
				
				// 범위를 벗어나거나 다음 칸에 이미 값이 채워졌다면 방향을 바꾼다. 
				if(nx>=N || ny>=N || nx<0 || ny<0 || arr[nx][ny]>0) {
					dir = (dir+1)%4;
				}
				
				x+=dx[dir];
				y+=dy[dir];
			}
		
			bw.write("#"+t+ "\n");
			for(int i=0; i<N; i++) {
				for(int j=0; j<N; j++) {
					bw.write(arr[i][j] + " ");
				}
				bw.write("\n");
			}
		}
		bw.flush();
		bw.close();
	}
}