[SWEA][JAVA] 1954. 달팽이 숫자

 

class Solution
{
    public static void main(String args[]) throws Exception
    {
        Scanner sc = new Scanner(System.in);
        int T;
        T=sc.nextInt();
		
        int [] r = {1,0,-1,0}; // 3, 6, 9, 12
        int [] c= {0, 1, 0,-1};
        int cur = 0;
        for(int test_case = 1; test_case <= T; test_case++)
        {
            int n = sc.nextInt();
            int[][] map = new int[n][n];
            int x=0; int y=0;
            for(int i=1; i<=n*n;i++){
                map[x][y] = i;
                if(i == n || i == n+(n-1) ||i == n+(n-1)+(n-1)){
                    // 방향 틀기
                    cur++;
                }
                x+= c[cur];
                y+= r[cur];
            }
            StringBuilder sb = new StringBuilder();
            for(int i=0;i<n;i++){
                for(int j=0;j<n;j++){
                    sb.append(map[i][j]).append(" ");
                }
                sb.append("\n");
            }
            System.out.printf("#%d\n%s",test_case, sb.toString());
        }
    }
}

 

---

오답노트(푸는 동안 사고과정 및 개선점)

꼭지점에서 회전시키고, 델타 배열을 이용해서 방향전환을 하게 했다. 문제는 네 꼭지점뿐만 아니라 방향을 트는 조건이 한바퀴를 돌때마다 달라진다는 것이다. 이걸 제대로 작성하려면 진행방향 +1에 값이 채워져있으면 방향을 돌려버리는 로직이 필요하다.

---

정답코드

class Solution
{
    public static void main(String args[]) throws Exception
    {
        Scanner sc = new Scanner(System.in);
        int T;
        T=sc.nextInt();
	
        int [] r = {1,0,-1,0}; // 12, 3, 6, 9
        int [] c= {0, 1, 0,-1};
        for(int test_case = 1; test_case <= T; test_case++)
        {
            int cur = 0;
            int n = sc.nextInt();
            int[][] map = new int[n][n];
            int x=0; int y=0;
            
            for(int i=1; i<=n*n;i++){
                map[x][y] = i;

                if(x+c[cur%4] >= n || y+r[cur%4] >= n ||  x+c[cur%4] <0 || y+r[cur%4] <0){
                    // 방향 틀기
                    cur++;
                    x+= c[cur%4];
                    y+= r[cur%4];
                }else{
                    if(map[x+c[cur%4]][y+r[cur%4]] != 0){
                        cur++;
                        x+= c[cur%4];
                        y+= r[cur%4];
                    }else{
                        x+= c[cur%4];
                        y+= r[cur%4];
                    }
                }
            }
            StringBuilder sb = new StringBuilder();
            for(int i=0;i<n;i++){
                for(int j=0;j<n;j++){
                    sb.append(map[i][j]).append(" ");
                }
                sb.append("\n");
            }
            System.out.printf("#%d\n%s",test_case, sb.toString());
        }
    }
}

 

조건을 까다롭게 필터링해서 방향전환을 시켰다.

IndexOutOfBound 안나게 조심하면 쉽게 풀린다. 배열 회전을 경험할 수 있는 좋은 문제같다.

 

도움이 됐다면 댓글이나 공감 버튼 한 번씩 누르고 가주세요! 로그인 안해도 됩니다 ^_^