[JAVA]68645 삼각달팽이

 📚 문제  

프로그래머스

 

🔍 아이디어  

만약 n이 4라면 아래로 4칸 -> 오른쪽으로 3칸 -> 위로 2칸 -> 아래로 1칸 과 같은 방식으로 움직이는 것을 확인했습니다.

또한 아래로 내려가는경우 col은 동일하고 row 만 달라지고, 오른쪽으로만 이동하는 경우는 col만, 위쪽으로 이동하는 경우만 row, col을 모두 줄여주면서 진행하면 됩니다. 

하지만 만약 포문중 마지막에 움직이면서 해당 영향으로 다음 row, col이 변화하기 때문에 마지막 칸은 다음칸의 움직임으로 이동하였습니다.(아래의 색칠된 부분이 동일한 것을 확인할 수 있습니다.)

if(i<n) row++;                 else col++;

if(i<n) col++;                 else {                     row--;                     col--;   }

if(i<n) {                     row--;                     col--;                 }else{                     row++;                 }

 

📝 풀이 

import java.util.*;
import java.io.*;
class Solution {
    public int[] solution(int n) {
        int[][] answer = new int[n][n];
        int row = 0, col = 0;
        int now = 1;
        while(0<n){
            //아래로 내려가기
            for(int i=1;i<=n;i++){
                answer[row][col] = now++;
                if(i<n) row++;
                else col++;
            }
            n--;
            
            //왼쪽 이동
            for(int i=1;i<=n;i++){
                answer[row][col] = now++;
                if(i<n) col++;
                else {
                    row--;
                    col--;
                }
            }
            n--;
            
            //위로 올라가기
            for(int i=1;i<=n;i++){
                answer[row][col] = now++;
                if(i<n) {
                    row--;
                    col--;
                }else{
                    row++;
                }
          
            }
            n--;   
        }
        
        int[] ans = new int[(answer.length)*(answer.length+1)/2];
        int index = 0;
        for(int i=0;i<answer.length;i++){
            for(int j=0;j<i+1;j++){
                ans[index++] = answer[i][j];
            }
        }
        
        return ans;
    }
}