refactor: WordLadder
(#5434)
* refactor: WordLadder * refactor: fix redundant check --------- Co-authored-by: alxkm <alx@alx.com>
This commit is contained in:
parent
87cf89192b
commit
14916e692f
@ -4,58 +4,28 @@ import java.util.HashSet;
|
|||||||
import java.util.LinkedList;
|
import java.util.LinkedList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Queue;
|
import java.util.Queue;
|
||||||
|
import java.util.Set;
|
||||||
|
|
||||||
/*
|
/**
|
||||||
**Problem Statement:**
|
* Class to find the shortest transformation sequence from a beginWord to an endWord using a dictionary of words.
|
||||||
A transformation sequence from word beginWord to word endWord using a dictionary wordList is a
|
* A transformation sequence is a sequence of words where each adjacent pair differs by exactly one letter.
|
||||||
sequence of words beginWord -> s1 -> s2 -> ... -> sk such that:
|
|
||||||
|
|
||||||
Every adjacent pair of words differs by a single letter.
|
|
||||||
Every si for 1 <= i <= k is in wordList. Note that beginWord does not need to be in wordList.
|
|
||||||
sk == endWord
|
|
||||||
Given two words, beginWord and endWord, and a dictionary wordList, return the number of words in
|
|
||||||
the shortest transformation sequence from beginWord to endWord, or 0 if no such sequence exists.
|
|
||||||
|
|
||||||
**Example 1:**
|
|
||||||
Input: beginWord = "hit", endWord = "cog", wordList = ["hot","dot","dog","lot","log","cog"]
|
|
||||||
Output: 5
|
|
||||||
Explanation: One shortest transformation sequence is "hit" -> "hot" -> "dot" -> "dog" -> cog",
|
|
||||||
which is 5 words long.
|
|
||||||
|
|
||||||
**Example 2:**
|
|
||||||
Input: beginWord = "hit", endWord = "cog", wordList = ["hot","dot","dog","lot","log"]
|
|
||||||
Output: 0
|
|
||||||
Explanation: The endWord "cog" is not in wordList, therefore there is no valid transformation
|
|
||||||
sequence.
|
|
||||||
|
|
||||||
**Constraints:**
|
|
||||||
1 <= beginWord.length <= 10
|
|
||||||
endWord.length == beginWord.length
|
|
||||||
1 <= wordList.length <= 5000
|
|
||||||
wordList[i].length == beginWord.length
|
|
||||||
beginWord, endWord, and wordList[i] consist of lowercase English letters.
|
|
||||||
beginWord != endWord
|
|
||||||
All the words in wordList are unique.
|
|
||||||
*/
|
*/
|
||||||
|
public final class WordLadder {
|
||||||
final class WordLadder {
|
|
||||||
private WordLadder() {
|
private WordLadder() {
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* This function finds the ladderLength
|
* Finds the shortest transformation sequence from beginWord to endWord.
|
||||||
*
|
*
|
||||||
* @param beginWord: Starting word of the ladder
|
* @param beginWord the starting word of the transformation sequence
|
||||||
* @param endWord: Ending word of the ladder
|
* @param endWord the target word of the transformation sequence
|
||||||
* @param wordList: This list contains the words which needs to be included
|
* @param wordList a list of words that can be used in the transformation sequence
|
||||||
* in ladder.
|
* @return the number of words in the shortest transformation sequence, or 0 if no such sequence exists
|
||||||
* @return ladderLength: This function will return the ladderLength(level)
|
|
||||||
* if the endword is there. Otherwise, will return the length as 0.
|
|
||||||
*/
|
*/
|
||||||
public static int ladderLength(String beginWord, String endWord, List<String> wordList) {
|
public static int ladderLength(String beginWord, String endWord, List<String> wordList) {
|
||||||
HashSet<String> set = new HashSet<>(wordList);
|
Set<String> wordSet = new HashSet<>(wordList);
|
||||||
|
|
||||||
if (!set.contains(endWord)) {
|
if (!wordSet.contains(endWord)) {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -66,25 +36,25 @@ final class WordLadder {
|
|||||||
while (!queue.isEmpty()) {
|
while (!queue.isEmpty()) {
|
||||||
int size = queue.size();
|
int size = queue.size();
|
||||||
for (int i = 0; i < size; i++) {
|
for (int i = 0; i < size; i++) {
|
||||||
String curr = queue.poll();
|
String currentWord = queue.poll();
|
||||||
char[] wordsChars = curr.toCharArray();
|
char[] currentWordChars = currentWord.toCharArray();
|
||||||
for (int j = 0; j < wordsChars.length; j++) {
|
for (int j = 0; j < currentWordChars.length; j++) {
|
||||||
char originalChars = wordsChars[j];
|
char originalChar = currentWordChars[j];
|
||||||
for (char c = 'a'; c <= 'z'; c++) {
|
for (char c = 'a'; c <= 'z'; c++) {
|
||||||
if (wordsChars[j] == c) {
|
if (currentWordChars[j] == c) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
wordsChars[j] = c;
|
currentWordChars[j] = c;
|
||||||
String transformedWord = String.valueOf(wordsChars);
|
String newWord = new String(currentWordChars);
|
||||||
if (transformedWord.equals(endWord)) {
|
|
||||||
|
if (newWord.equals(endWord)) {
|
||||||
return level + 1;
|
return level + 1;
|
||||||
}
|
}
|
||||||
if (set.contains(transformedWord)) {
|
if (wordSet.remove(newWord)) {
|
||||||
set.remove(transformedWord);
|
queue.offer(newWord);
|
||||||
queue.offer(transformedWord);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
wordsChars[j] = originalChars;
|
currentWordChars[j] = originalChar;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
level++;
|
level++;
|
||||||
|
@ -5,6 +5,8 @@ import static org.junit.jupiter.api.Assertions.assertEquals;
|
|||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.Test;
|
||||||
|
import org.junit.jupiter.params.ParameterizedTest;
|
||||||
|
import org.junit.jupiter.params.provider.CsvSource;
|
||||||
|
|
||||||
public class WordLadderTest {
|
public class WordLadderTest {
|
||||||
|
|
||||||
@ -38,4 +40,12 @@ public class WordLadderTest {
|
|||||||
List<String> wordList2 = Arrays.asList("hot", "dot", "dog", "lot", "log");
|
List<String> wordList2 = Arrays.asList("hot", "dot", "dog", "lot", "log");
|
||||||
assertEquals(WordLadder.ladderLength("hit", "cog", wordList2), 0);
|
assertEquals(WordLadder.ladderLength("hit", "cog", wordList2), 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ParameterizedTest
|
||||||
|
@CsvSource({"'a', 'c', 'b,c', 2", "'a', 'c', 'a', 0", "'a', 'a', 'a', 0", "'ab', 'cd', 'ad,bd,cd', 3", "'a', 'd', 'b,c,d', 2", "'a', 'd', 'b,c,e,f,g,h,i,j,k,l,m,n,o,p,q,r,s,t,u,v,w,x,y,z,d', 2"})
|
||||||
|
void testLadderLength(String beginWord, String endWord, String wordListStr, int expectedLength) {
|
||||||
|
List<String> wordList = List.of(wordListStr.split(","));
|
||||||
|
int result = WordLadder.ladderLength(beginWord, endWord, wordList);
|
||||||
|
assertEquals(expectedLength, result);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user