프로그래머스 - 영어 끝말잇기

https://programmers.co.kr/learn/courses/30/lessons/12981

코딩테스트 연습 - 영어 끝말잇기

 

 

 

 

 

문자열을 다루는 문제이다. 문자열에서 문자를 뽑아내기 위해서 str.charAt(i)과 같은 메소드가 필요하다.

그리고 문자열 배열의 길이 vs 문자열의 길이

words.length vs word.length()

 

자료구조는 ArrayList를 사용하여 중복을 체크하도록 했다.

 

+ 다른 사람의 풀이를 보니 HashSet을 사용했다.

HashSet : 중복을 자동으로 제거해서 저장한다.

 

https://coding-factory.tistory.com/554

[Java] 자바 HashSet 사용법 & 예제 총정리

 

 

 

코드

import java.util.*;

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = {0, 0};

        ArrayList<String> list = new ArrayList<>();
        
        for(int i = 0; i < words.length; i++) {
            if(i >= 1) { //다른 글자로 시작
                char ch1 = words[i - 1].charAt(words[i - 1].length() - 1);
                char ch2 = words[i].charAt(0);
                
                if(ch1 != ch2) {
                    answer[0] = i % n + 1;
                    answer[1] = i / n + 1;
                    
                    break;
                }
            }
            
            
            if(!list.contains(words[i])) { //중복된 단어 말함
                list.add(words[i]);
            } else {
                answer[0] = i % n + 1;
                answer[1] = i / n + 1;
                
                break;
            }
        }

        return answer;
    }
}

 

 

 

 

 

결과