[백준] 1544 사이클 단어 - Data Structure / Java

• 문제 링크

1544번: 사이클 단어

 

• 풀이 과정

각 단어를 입력받을 때 해당 단어가 set에 존재하는 지 확인하고,

만약 존재하지 않다면 해당 단어로 만들어지는 모든 사이클 단어를 구해 set에 저장한다.

사이클 단어는 메모리 효율을 위해 StringBuilder 를 활용해 단어의 앞뒤를 순환시켜 구할 수 있으며,

모든 사이클 단어를 set에 저장했을 시 카운트를 증가시킨다.

 

이 후 특정 단어의 사이클 단어가 입력될 시 이미 해당 단어는 포함하므로 건너뛰고,

새로운 단어가 입력될 시 위의 과정을 반복해 서로 다른 단어의 수를 구한다.

 

• 풀이 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.HashSet;
import java.util.Set;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int n = Integer.parseInt(br.readLine());

        Set<String> set = new HashSet<>();
        int cnt = 0;

        while (n-- > 0) {
            String input = br.readLine();
            if (!set.contains(input)) {
                StringBuilder sb = new StringBuilder(input);
                for (int i = 0; i < input.length(); i++)
                    set.add(sb.append(sb.charAt(0)).deleteCharAt(0).toString());
                cnt++;
            }
        }

        bw.write(String.valueOf(cnt));
        bw.flush();
    }

}