[Silver II] Title: 기차가 어둠을 헤치고 은하수를, Time: 596 ms, Memory: 82172 KB -BaekjoonHub

Author: shimhun99

백준/Silver/15787. 기차가 어둠을 헤치고 은하수를/README.md

@@ -4,15 +4,15 @@
 
 ### 성능 요약
 
-메모리: 75440 KB, 시간: 600 ms
+메모리: 82172 KB, 시간: 596 ms
 
 ### 분류
 
 비트마스킹, 구현
 
 ### 제출 일자
 
-2025년 1월 24일 15:33:43
+2025년 1월 24일 15:44:30
 
 ### 문제 설명
 

백준/Silver/15787. 기차가 어둠을 헤치고 은하수를/기차가 어둠을 헤치고 은하수를.java

@@ -10,54 +10,106 @@ public static void main(String[] args) throws IOException {
         int N = Integer.parseInt(split[0]);
         int M = Integer.parseInt(split[1]);
 
-        // 기차 배열 초기화
-        int[][] arr = new int[N + 1][21];
+        /*
+         * 기차에 사람이 타있으면 1, 없으면 0
+         * */
 
-        // 명령 처리
-        for (int i = 0; i < M; i++) {
-            String[] line = reader.readLine().split(" ");
-            int train = Integer.parseInt(line[1]);
+        int[][] arr = new int[N+1][21];
 
-            switch (line[0]) {
-                case "1": // 승차
-                    int seatToBoard = Integer.parseInt(line[2]);
-                    arr[train][seatToBoard] = 1;
+        for(int i=0; i<M; i++){ //M개의 명령 실행
+            String[] line = reader.readLine().split(" ");
+            int train_i = Integer.parseInt(line[1]);
+//            System.out.println(line[0]);
+            switch (line[0]){
+                case "1" :
+                    first(arr, train_i, Integer.parseInt(line[2]));
                     break;
-                case "2": // 하차
-                    int seatToLeave = Integer.parseInt(line[2]);
-                    arr[train][seatToLeave] = 0;
+                case "2" :
+                    second(arr, train_i, Integer.parseInt(line[2]));
                     break;
-                case "3": // 뒤로 이동
-                    for (int j = 20; j > 1; j--) { // 20번 칸까지 뒤로 이동
-                        arr[train][j] = arr[train][j - 1];
-                    }
-                    arr[train][1] = 0; // 1번 좌석은 빈 상태
+                case "3" :
+                    third(arr, train_i);
                     break;
-                case "4": // 앞으로 이동
-                    for (int j = 1; j < 20; j++) { // 1번 칸부터 앞으로 이동
-                        arr[train][j] = arr[train][j + 1];
-                    }
-                    arr[train][20] = 0; // 20번 좌석은 빈 상태
+                case "4" :
+                    fourth(arr, train_i);
                     break;
                 default:
-                    // 잘못된 명령어는 무시
                     break;
             }
         }
 
-        // 중복된 상태 제거
+//        for(int[] array : arr){
+//            for(int num : array){
+//                System.out.print(num + " ");
+//            }
+//            System.out.println();
+//        }
+
         HashSet<String> set = new HashSet<>();
-        for (int i = 1; i <= N; i++) {
-            StringBuilder sb = new StringBuilder();
-            for (int j = 1; j <= 20; j++) { // 좌석 상태만 문자열로 변환
-                sb.append(arr[i][j]);
-            }
-            set.add(sb.toString()); // 중복된 배열 상태 제거
+        //배열들을 map에 넣는다면?
+//        for(int i=1; i<N+1; i++){
+//            set.add(arr[i]);
+//        }
+
+//        for(int[] array : arr){
+//            String s = Arrays.toString(array);
+//            set.add(s);
+//        }
+
+//        for (int i = 1; i <= N; i++) { // i = 1부터 시작
+//            for(int j=1; j<21; j++){
+//                System.out.print(arr[i][j] + " ");
+//            }
+//            System.out.println();
+//        }
+
+        for (int i = 1; i <= N; i++) { // i = 1부터 시작
+            String s = Arrays.toString(arr[i]);
+            set.add(s);
         }
 
-        // 결과 출력
-        writer.write(set.size() + "\n");
-        writer.flush();
-        writer.close();
+        System.out.println(set.size());
+
+    }
+
+    public static void first(int[][] arr, int i, int x){ //해당 칸 승차
+//        System.out.println("test1");
+
+//        if(arr[i][x] == 0){
+        arr[i][x] = 1;
+//        }
+    }
+
+    public static void second(int[][] arr, int i, int x){ //해당 칸 하차
+//        System.out.println("test2");
+
+//        if(arr[i][x] == 1){
+        arr[i][x] = 0;
+//        }
+    }
+
+    public static void third(int[][] arr, int i){ //한칸 씩 뒤로 이동
+//        System.out.println("test3");
+        for(int j=20; j>1; j--){
+            arr[i][j] = arr[i][j-1];
+//            if(arr[i][j] == 1){ // 해당 좌석에 사람이 있는 경우에만
+//                arr[i][j+1] = 1; // 뒤로 좌석 옮기기
+//                arr[i][j] = 0;
+//            }
+        }
+        arr[i][1] = 0;
+    }
+
+    public static void fourth(int[][] arr, int i){ //한칸 씩 앞으로 이동
+//        System.out.println("test4");
+
+        for(int j=1; j<20; j++){
+            arr[i][j] = arr[i][j+1];
+//            if(arr[i][j] == 1){ // 해당 좌석에 사람이 있는 경우에만
+//                arr[i][j-1] = 1; //앞으로 좌석 옮기기
+//                arr[i][j] = 0;
+//            }
+        }
+        arr[i][20] = 0;
     }
 }