[프로그래머스/Lv. 2] 영어 끝말잇기

🔺 문제

프로그래머스

 

🔺 코드

- 틀림

import java.util.*;

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

        Map<String, Integer> map = new HashMap<>();
        map.put(words[0], 1);
        
        for(int i = 1 ; i < words.length - 1 ; i++) {
            if(map.containsKey(words[i]) || words[i-1].charAt(words[i-1].length() - 1) != words[i+1].charAt(0)) {
                answer[0] = (i % n) + 1; // 사람 (8 % 3 + 1)
                answer[1] = (i / n) + 1; // 차례 (8 / 3 + 1)
                break;        
            }
            
            map.put(words[i], 1);
        }

        return answer;
    }
}

HashMap을 써볼까 생각했었다...  로직이 틀린 거겠지,,,

 

 

- 정답 1) List 활용

import java.util.*;

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = new int[2];
        List<String> list = new ArrayList<>();
      
        for(int i = 0 ; i < words.length ; i++) {
            if(list.contains(words[i])) {
                answer[0] = i % n + 1; // 사람
                answer[1] = i / n + 1; // 차례
                break;
            }
        
            list.add(words[i]);
        
            if(i > 0 && words[i-1].charAt(words[i-1].length() - 1) != words[i].charAt(0)) {            
                answer[0] = (i % n) + 1;  // 사람
                answer[1] = (i / n) + 1;  // 차례
                break;
            }
        }
        return answer;
    }
}

✅ 해결 아이디어
- 리스트 활용.
- for문 안에서 조건문
└ 첫 번째  if문 : 중복되는 단어인 경우
└ 두 번째 if문 : (이전 단어의 마지막 글자 ≠ 현재 단어의 첫 번째 글자)인 경우

 

 

- 정답 2) Set 활용

import java.util.*;

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = new int[2];  // 사람, 차례
        
        Set<String> set = new HashSet<>();
        set.add(words[0]);
        
        for(int i = 1 ; i < words.length ; i++) {
            if(!set.contains(words[i]) && words[i-1].charAt(words[i-1].length() -1) == words[i].charAt(0)) {
                set.add(words[i]);
            } else {
                answer[0] = (i % n) + 1;  // 사람
                answer[1] = (i / n) + 1;  // 차례
                break;
            }
        }

        return answer;
    }
}

 

---

🔺 다른 풀이들

- 풀이1) HashSet 사용

프로그래머스

import java.util.HashSet;

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = new int[2];
        HashSet<String> set = new HashSet<String>();
        set.add(words[0]);
        
        String prev = "";

        for(int i = 1; i < words.length; i ++){
            prev = words[i - 1];
            
            if(set.contains(words[i]) || prev.charAt(prev.length() - 1) != words[i].charAt(0)){
                answer[0] = i % n + 1;	// 사람
                answer[1] = i / n + 1;	// 차례
                break;
            }
            set.add(words[i]);
        }
        return answer;
    }
}

HashSet을 이용하셨다. (HashSet - 중복 제거) 

words[0]은 미리 set에 넣어놓고, 1부터 시작하게 하셨다.

중복 체크할 때는 contains() 이용하심.

이게 제일 깔끔한 풀이 같다.

 

 

-풀이2) boolean 사용 (+ 리스트)

프로그래머스[Java] - (Level2)영어 끝말잇기

여기도 중복 체크할 때 contains()를 사용하셨다.

중복인 경우, boolean flag를 false로 값 변경하는 식으로 하셨다.

 

 

- 풀이3) HashMap 사용

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

내가 사용하려고 생각했던 HashMap을 사용하셨다!

덕분에 내 로직이 오디서 잘못되었음을 확인하였음,,!! 

---

(참고)

- 풀이

[프로그래머스] 영어 끝말잇기 (JAVA)

 

- HashSet 사용법 (add, remove, size, contains)

[JAVA] HashSet이란? & 사용법 정리

 

+ 중복제거

중복체크및 제거(String, HashMap, HashSet, Stream)