Word Ladder

Given two words (beginWord and endWord), and a dictionary's word list, find the length of shortest transformation sequence from beginWord to endWord, such that:

Only one letter can be changed at a time. Each transformed word must exist in the word list. Note that beginWord is not a transformed word.

For example,

Given: beginWord = "hit" endWord = "cog" wordList = ["hot","dot","dog","lot","log","cog"]

As one shortest transformation is "hit" -> "hot" -> "dot" -> "dog" -> "cog", return its length 5.

Note:

Return 0 if there is no such transformation sequence. All words have the same length. All words contain only lowercase alphabetic characters. You may assume no duplicates in the word list. You may assume beginWord and endWord are non-empty and are not the same.

UPDATE (2017/1/20): The wordList parameter had been changed to a list of strings (instead of a set of strings). Please reload the code definition to get the latest changes.

Solution

public class Solution {
          public int ladderLength(String beginWord, String endWord, Set<String> wordList) {
        Map<String, Integer> map = new HashMap<>();
        Queue<String> queue = new LinkedList<String>();
        queue.add(beginWord);
        map.put(beginWord, 1);

        while (!queue.isEmpty()) {
            String currentString = queue.poll();
            if (currentString.equals(endWord)) {
                break;
            }

            for (int i = 0; i < currentString.length(); i++) {
                char[] charArray = currentString.toCharArray();

                for (char j = 'a'; j <= 'z'; j++) {
                    if (j != currentString.charAt(i)) {
                        charArray[i] = j;

                        String newString = String.valueOf(charArray);
                        if (wordList.contains(newString) && !map.containsKey(newString)) {
                            map.put(newString, map.get(currentString) + 1);
                            queue.add(newString);
                        }
                    }
                }

                charArray[i] = currentString.charAt(i);
            }
        }

        return map.containsKey(endWord) ? map.get(endWord) : 0;
    }
}