[ 백준 / BOJ 22861 ] 폴더 정리 - large ( JAVA / 자바 )

https://www.acmicpc.net/problem/22861

22861번: 폴더 정리 (large)

문제

이름이 main 폴더 안에 여러가지 파일과 폴더가 존재한다.

main
 ├─ FolderA
 │    ├─ File1
 │    └─ File2
 └─ FolderB
       ├─ FolderC
       │   ├─ File4
       │   └─ File5
       ├─ File1
       └─ File3

위 구조는 main 폴더의 하위 구조를 계층적으로 표시한 것이다. FolderA, FolderB, FolderC는 폴더이고 File1, File2, File3은 파일이다. 파일 이름이 같은 경우는 내용이 완전 동일한 파일이다.

한 폴더 안에 같은 이름을 가진 파일이 두 개 이상 존재할 수 없다.

main 폴더의 하위 디렉토리에 같은 이름의 폴더가 두 개 이상 존재할 수 없다.

폴더 또는 파일을 옮겨 main 폴더를 정리하려고 한다.

예를 들어, FolderB 폴더 하위에 있는 두 개의 파일 File1, File3과 하나의 폴더 FolderC를 FolderA 폴더 하위에 옮기려고 한다.

File1 파일은 FolderA 폴더 안에 동일한 파일이 이미 존재하므로 덮어쓴다. File3은 동일한 파일이 없으므로 그대로 옮긴다. FolderC 폴더도 동일한 폴더가 없으므로 FolderA 폴더 하위에 옮긴다. FolderB 폴더 하위에 있는 폴더와 파일을 다 옮겼으므로 FolderB는 삭제한다.

아래는 FolderB 폴더를 FolderA 폴더 안에 옮긴 후 main 폴더의 하위 구조를 계층적으로 표시한 것이다.

main
 └─ FolderA
      ├─ FolderC
      │ ├─ File4
      │ └─ File5
      ├─ File1
      ├─ File2
      └─ File3

이러한 과정을 통해 폴더를 정리하려고 한다. 폴더 정리 후 쿼리를 통하여 파일의 정보를 확인하려고 한다.

입력

첫 번째 줄에는 main 폴더 안에 있는 폴더의 총 개수 N과 파일의 총 개수 M이 공백으로 구분되어 주어진다.

두 번째 줄부터 N+M+1번째까지 상위 폴더의 이름 P, 폴더 또는 파일의 이름 F, 폴더인지 아닌지 알려주는 C가 공백으로 구분되어 주어진다.

 C의 값은 F가 폴더라면 1, 파일이라면 0으로 주어진다.

 N+M+2번째 줄에는 옮기는 횟수 K가 주어진다.

그 다음 줄부터 총 K 줄에 걸쳐 폴더 경로 A와 폴더 경로 B가 공백으로 구분되어 주어진다.

옮기는 행위는 입력 순서대로 수행이 되어야 한다.

 A 하위에 있는 파일과 폴더들을 B하위에 옮기는 것이다. 이때, A는 B의 상위 폴더가 아님을 보장한다.

그 다음 줄에는 쿼리의 개수 Q가 주어진다.

그 다음 줄부터 Q개의 쿼리가 주어진다. 쿼리마다 main으로부터 폴더의 경로 정보가 들어온다. 예를 들어 main 폴더 안에 FolderB에 대한 쿼리가 들어온다면, FolderB의 경로인 main/FolderB로 주어진다. 반드시 폴더가 존재하는 경로로 주어짐을 보장한다.

출력

쿼리 순서대로 한 줄씩 폴더 하위에 있는 파일의 종류의 개수와 파일의 총 개수를 출력한다.

파일의 종류의 개수는 같은 파일이 여러개 있을 경우 하나로 계산한다. 파일의 총 개수는 같은 파일이 있더라도 하나로 계산하지 않는다.

예를 들어 이름이 File1 파일이 5개가 있을 경우 파일의 종류는 1 가지이고 파일의 총 개수는 5개이다.

제한

• $1 \le N \le 1,000$ 
• $1 \le M \le 1,000$ 
• $0 \le K \le 1,000$ 
• $1 \le |P| \le 10$ 
• $1 \le |F| \le 10$ 
• $0 \le C \le 1$ 
• $1 \le Q \le 1,000$

풀이

1) 입력받기

Input(String parFolder, String name, int isWhat)

parFolder : 부모 폴더 이름
name : 폴더 또는 파일 이름
isWhat : 폴더인지 파일인지 여부 (0 → 파일, 1 → 폴더)

1-1) 파일이라면,
HashMap<String, HashSet<String>> haveFiles
parFolder를 key로 name 추가

1-2) 폴더라면,
HashMap<String, HashSet<String>> haveFolders
parFolder를 key로 name 추가

✅ value를 HashSet으로 둔 이유는, 같은 폴더내에는 같은 이름의 파일 및 폴더가 존재할 수 없기 때문 !!

2) 이동하기

move(String[] folderA, String[] folderB)

folderA의 하위 폴더와 파일들을 folderB로 옮기는 메서드
간단하게
haveFolders,haveFiles에 folderA가 key로 존재한다면 👉 하위에 파일, 폴더가 존재한다는 의미
set의 값들을 folderB 를 key로 추가
그 후, folderA는 haveFolders와 haveFiles에서 삭제

3) 하위 파일 갯수와 종류 찾기

find(String[] folders)

• 변수
  cnt : 종류의 갯수
  HashSet<String> set : 파일을 담기 위한 set (✅ 중복되면 안되기 때문에)

findFolderAndFile(target) 메서드 호출
👉 만약 target 하위에 폴더가 있다면, findFolderAndFile(next) 호출
👉 만약 target 하위에 파일이 있다면, set에 file 추가하고, cnt+=1

끝나면, cnt와 set의 크기 출력

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main_BOJ_22861_폴더정리_large {

    static int N, M, K, Q, cnt = 0;
    static final int FILE = 0, FOLDER = 1;
    static Map<String, HashSet<String>> haveFiles = new HashMap<>();
    static Map<String, HashSet<String>> haveFolders = new HashMap<>();
    static StringBuilder sb = new StringBuilder();
    static Set<String> set;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer stringTokenizer = new StringTokenizer(br.readLine());

        N = Integer.parseInt(stringTokenizer.nextToken());
        M = Integer.parseInt(stringTokenizer.nextToken());

        for (int i = 0; i < N + M; i++) {
            stringTokenizer = new StringTokenizer(br.readLine());
            String parFolder = stringTokenizer.nextToken();
            String name = stringTokenizer.nextToken();
            int isWhat = Integer.parseInt(stringTokenizer.nextToken());

            Input(parFolder, name, isWhat);
        }

        K = Integer.parseInt(br.readLine());
        while (K-- > 0) {
            stringTokenizer = new StringTokenizer(br.readLine());
            String[] folderA = stringTokenizer.nextToken().split("/");
            String[] folderB = stringTokenizer.nextToken().split("/");

            move(folderA, folderB);
        }

        Q = Integer.parseInt(br.readLine());
        while (Q-- > 0) {
            stringTokenizer = new StringTokenizer(br.readLine());
            String[] folders = stringTokenizer.nextToken().split("/");

            find(folders);
        }

        System.out.print(sb);
    }

    private static void find(String[] folders) {
        cnt = 0;
        set = new HashSet<>();
        String target = folders[folders.length - 1];
        findFolderAndFile(target);

        sb.append(cnt + " " + set.size()).append("\n");
    }

    private static void findFolderAndFile(String target) {
        if (haveFolders.containsKey(target)) {
            for (String nextFolder : haveFolders.get(target)) {
                findFolderAndFile(nextFolder);

            }
        }
        if (haveFiles.containsKey(target)) {
            for (String file : haveFiles.get(target)) {
                set.add(file);
                cnt += 1;
            }
        }
    }

    private static void move(String[] folderA, String[] folderB) {
        String target = folderA[folderA.length - 1];
        String dest = folderB[folderB.length - 1];

        if (haveFolders.containsKey(target)) {
            if(!haveFolders.containsKey(dest)) haveFolders.put(dest, new HashSet<>());

            for (String folder : haveFolders.get(target)) {
                haveFolders.get(dest).add(folder);
            }

            haveFolders.remove(target);
        }

        if (haveFiles.containsKey(target)) {
            if(!haveFiles.containsKey(dest)) haveFiles.put(dest, new HashSet<>());

            for (String file : haveFiles.get(target)) {
                haveFiles.get(dest).add(file);
            }

            haveFiles.remove(target);
        }

        String parent = folderA[folderA.length - 2];
        haveFolders.get(parent).remove(target);

    }

    private static void Input(String parFolder, String name, int isWhat) {
        if (isWhat == FOLDER) {
            if(!haveFolders.containsKey(parFolder)) haveFolders.put(parFolder, new HashSet<>());

            haveFolders.get(parFolder).add(name);
        } else if (isWhat == FILE) {
            if(!haveFiles.containsKey(parFolder)) haveFiles.put(parFolder, new HashSet<>());

            haveFiles.get(parFolder).add(name);
        }
    }
}