[PGS] 삼각달팽이

Programmers/Gayo/삼각달팽이.java

@@ -0,0 +1,36 @@
+class Solution {
+    public int[] solution(int n) {
+        int[] answer = {};
+        int[][] arr = new int[n][n];
+
+        int x = -1, y = 0;
+        int num = 1;
+        for(int i=0; i<n; i++){
+            if(i%3==0){
+                for(int j=0; j<n-i; j++){
+                    arr[++x][y] = num++;
+                }
+            }
+            else if(i%3==1){
+                for(int j=0; j<n-i; j++){
+                    arr[x][++y] = num++;
+                }
+            }
+            else{
+                for(int j=0; j<n-i; j++){
+                    arr[--x][--y] = num++;
+                }
+            }
+        }
+
+        answer = new int[num-1];
+        int idx = 0;
+        for(int i=0; i<arr.length; i++){
+            for(int j=0; j<=i; j++){
+                answer[idx++] = arr[i][j];
+            }
+        }
+
+        return answer;
+    }
+}