Java 백준 2866 문자열 잘라내기

백준 링크

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

2866번: 문자열 잘라내기

---

문제

입력

출력

입출력 예시

 

---

제출한 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.StringTokenizer;

public class Main {
	static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
	static StringTokenizer st;
	static StringBuilder sb;
	static int R, C;

	static StringBuilder[] StringData;

	static char[] charData;

	static HashSet<String> answer = new HashSet<>();

	public static void main(String[] args) throws NumberFormatException, IOException {
		
		// R, C 입력받기
		st = new StringTokenizer(br.readLine());
		
		R = Integer.parseInt(st.nextToken());
		C = Integer.parseInt(st.nextToken());
		// 데이터를 저장할 Char[] 선언
		StringData = new StringBuilder[C];
		charData = new char[C];
		// 각 열의 데이터를 저장할 StringBuilder를 각각 초기화
		for (int i = 0; i < C; i++) {
			StringData[i] = new StringBuilder();
		}
		// 1행의 데이터를 각 열에 해당하는 index의 StringBuilder에 추가
		for (int x = 0; x < R; x++) {
			charData = br.readLine().toCharArray();
			for (int y = 0; y < C; y++) {
				StringData[y].append(charData[y]);
			}
		}

		int cnt = 0;
		int start = 0;
		// 로직 수행
		while (start++ < R) {
			
			for (int i = 0; i < C; i++) {
				// 0번재 index의 데이터 삭제
				StringData[i] = StringData[i].deleteCharAt(0);
				// 존재하는지 비교
				// 존재하는 경우 return
				if (answer.contains(StringData[i].toString())) {
					System.out.println(cnt);
					return;
				}
				// 존재하지 않는 경우 제일 처음 위치하는 데이터를 삭제하고 다시 저장
				else {
					answer.add(StringData[i].toString());
				}

			}
			answer.clear();
			cnt += 1;	
		}
		
		System.out.println(cnt);

	}

}

 

 

각 데이터를 열단위로 쪼개서 저장하고 비교해보는 방식으로 해결했다.

 

1. 입력받는 행의 데이터를 열 단위로 저장하기 위한 StringBuilder[]를 열의 길이만큼 할당한다.

2. 입력받는 데이터를 Char단위로 쪼개서 해당하는 위치의 StringBuilder에 저장한다.

3. 기본적으로 행의 개수만큼 while문을 반복하면서 로직을 수행한다.

4. StringBuilder[]에 저장된 StringBuilder의 첫번째 char를 제거한다.

5. 해당 데이터가 존재하지 않으면 빠른 속도로 존재하는지 확인하기 위한 HashSet에 저장하고 Hashset에 존재하는 경우 return한다.