diff --git a/src/problems/14.longest-common-prefix.jl b/src/problems/14.longest-common-prefix.jl index 057dd61b1..32ad7aeed 100644 --- a/src/problems/14.longest-common-prefix.jl +++ b/src/problems/14.longest-common-prefix.jl @@ -48,7 +48,7 @@ using LeetCode function longest_common_prefix(strs::Vector{String})::String s1, s2 = minimum(strs), maximum(strs) pos = findfirst(i -> s1[i] != s2[i], 1:length(s1)) - return isnothing(pos) ? "" : s1[1:(pos - 1)] + return isnothing(pos) ? s1 : s1[1:(pos - 1)] end ## @lc code=end diff --git a/test/problems/14.longest-common-prefix.jl b/test/problems/14.longest-common-prefix.jl index f21db06db..686a09db9 100644 --- a/test/problems/14.longest-common-prefix.jl +++ b/test/problems/14.longest-common-prefix.jl @@ -2,4 +2,7 @@ @test longest_common_prefix(["flower", "flow", "flight"]) == "fl" @test longest_common_prefix(["dog", "racecar", "car"]) == "" @test longest_common_prefix(["reflower", "flow", "flight"]) == "" + @test longest_common_prefix(["abc"]) == "abc" + @test longest_common_prefix(fill("abc", 5)) == "abc" + @test longest_common_prefix(fill("", 4)) == "" end