[프로그래머스] 영어 끝말잇기 (Summer/Winter Coding(~2018))

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

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

2021-07-22

---

import java.util.Arrays;
import java.util.HashSet;
 
public class Solution12981 {
    public static int[] solution(int n, String[] words) {
        HashSet<String> set = new HashSet<>();
        for(int i = 0; i < words.length; i++) {
            if(set.contains(words[i])) {
                return new int[]{i%n+1, i/n+1};
            } else {
                if(i > 0 && words[i-1].charAt(words[i-1].length()-1) != words[i].charAt(0)) return new int[]{i%n+1, i/n+1};
                set.add(words[i]);
            }
        }
        return new int[]{0,0};
    }
 
    public static void main(String[] args) {
        int n = 2;
        String[] w = {"hello", "one", "even", "never", "now", "world", "draw"};
        System.out.println(Arrays.toString(solution(n, w)));
    }
}

 

#문제풀이

1. HashSet을 이용해서 중복 체크를 해준다

 

2. 중복이 아닌 경우, 끝말잇기가 제대로 되는지 체크해준다.