214. Shortest Palindrome
Problem Description
In this problem, we are given a string s
. Our task is to make the smallest palindrome by adding characters at the front of s
. A palindrome is a string that reads the same forward and backward. The key here is that we can only add characters to the beginning of s
to form the palindrome; we cannot modify or add characters at the end. We need to find and return the shortest possible palindrome that can be obtained from s
through such transformations.
Intuition
To solve this problem effectively, we need to identify the longest palindrome that starts at the beginning of s
. Once we find such a palindrome, we can mirror the remaining part of the string (that isn't included in the palindrome) and add it to the front to create the shortest palindrome string.
The crucial part of the solution is to find the longest palindromic prefix efficiently. We can achieve this by comparing prefixes and suffixes of the string while respecting the palindrome property.
The algorithm uses a hash-based approach to compare the palindromic prefix and suffix, which can be efficiently computed using a rolling hash technique. Here's how it goes:
-
Initialize two hash variables
prefix
andsuffix
to 0. These will be used to calculate the hash of the prefix (from the start of the string) and the suffix (from the end of the string) respectively. -
Use a
base
for the hash function and amod
to prevent overflow issues with large hash values. Go over each character in the string and update the hash values for both the prefix and suffix. -
If at any point the hash of the prefix and the hash of the suffix are equal, it indicates that we have a palindrome from the beginning of the string to the current index
i
. -
Remember the furthest index
idx
where the prefix and suffix hashes match (indicating the longest palindromic prefix).
After we identify the longest palindromic prefix, the remaining substring (from idx
to the end), when reversed and added to the front of the original string s
, will result in the shortest palindrome.
So, the result will be the reverse of the substring from idx
to the end concatenated to the original string. If the entire string s
is a palindrome, we just return s
as it is already the shortest palindrome we can achieve.
Solution Approach
For the implementation of the solution, we leverage a rolling hash polynomial hashing algorithm. The rolling hash technique is an efficient way to compute and compare hash values for substrings quickly. This is particularly useful in our case, where we need to compare prefixes and suffixes of the given string.
Here's a step-by-step explanation of the implementation:
-
We choose a base number,
base
, for the rolling hash functions and a large prime number,mod
, to take the modulus and prevent integer overflow. -
We initialize
prefix
andsuffix
hash values to0
,mul
to1
, andidx
to0
. Theidx
will hold the position of the last character of the longest palindrome prefix. -
We iterate through the string
s
using a for loop, and on each iteration, we do the following:- Update the
prefix
hash by multiplying the currentprefix
hash bybase
and then adding the value of the current character shifted by1
to avoid0
in the calculation (ord(c) - ord('a') + 1
). We wrap around with modulusmod
to manage large numbers. - Update the
suffix
hash by adding to it the value of the current character multiplied bymul
(which representsbase
raised to the power of the character's position). Again, we take the result modulomod
. - Update
mul
by multiplying it bybase
and taking modulusmod
. This step effectively computesbase
to the power ofi
modulomod
for thei
-th character. - If at any point the
prefix
andsuffix
hashes are equal, it means we have a palindrome starting from the beginning of the string to the current character. We updateidx
toi + 1
.
- Update the
-
After the loop completes, if
idx
is equal to the length of the string, the entire string is a palindrome. We returns
in that case. -
If the whole string isn't a palindrome, we need to create a palindrome by adding characters to the beginning of the string. We do this by taking the substring from
idx
to the end ofs
, reversing it, and concatenating it with the original strings
.
The final result is generated by the expression s[idx:][::-1] + s
, where s[idx:][::-1]
creates the needed prefix by reversing the part of the string that is not included in the palindrome and appending it to the front of s
to form the shortest palindrome.
This approach is efficient as it has a linear time complexity with respect to the length of the string, and it leverages hashing to quickly identify the longest palindrome prefix.
Ready to land your dream job?
Unlock your dream job with a 2-minute evaluator for a personalized learning plan!
Start EvaluatorExample Walkthrough
Let's walk through the solution approach with a small example. Suppose our given string is s = "abacd"
. We want to find the shortest palindrome by adding characters at the beginning of s
.
-
We choose
base
as a small prime number, for simplicity, let's use3
, and letmod
be a large prime number,mod = 10007
to avoid integer overflow. -
Initialize
prefix
andsuffix
hashes to0
,mul
to1
, andidx
to0
. -
We'll iterate through the string and compute hashes for the prefix starting from the beginning and the suffix from the end simultaneously:
i Character (c) Prefix Hash (new) Suffix Hash (new) mul (new) idx 0 'a' (0*3 + 1) mod 10007 (0 + 1*1) mod 10007 3 1 1 'b' (1*3 + 2) mod 10007 (1 + 2*3) mod 10007 3*3 1 2 'a' (5*3 + 1) mod 10007 (7 + 1*9) mod 10007 333 3 3 'c' (16*3 + 3) mod 10007 (16 + 3*27) mod 10007 333*3 3 4 'd' (51*3 + 4) mod 10007 (97 + 4*81) mod 10007 33333 3 At each step, we are updating the
prefix
hash by multiplying it bybase
and adding the current character's value, and updating thesuffix
hash by adding the current character's value multiplied bymul
.mul
is updated by multiplying it bybase
at each step.At index
i = 2
, the prefix and suffix hashes match (16 mod 10007
), meaning that we have a palindrome from the start of the string to the character at index2
("aba"
). So we updateidx
to3
. -
After completing the iteration, since
idx
is not equal to the length of the string, we conclude that the entire string is not a palindrome. -
To form the shortest palindrome, we take the substring from
idx
to the end ("acd"
) and reverse it to get"dca"
. We then concatenate this reversed substring to the front of the original strings
.
The resulting palindrome is "dca" + "abacd"
= "dcaabacd"
, which is the shortest palindrome that we can form by adding characters only at the beginning of the string s
. This example illustrates the efficiency of the solution, which finds the longest palindromic prefix and adds the minimum required characters to the front to form the palindrome.
Solution Implementation
1class Solution:
2 def shortest_palindrome(self, s: str) -> str:
3 base = 131 # Base for polynomial rolling hash.
4 mod = 10**9 + 7 # Modulus for hash to avoid overflow.
5 n = len(s) # Length of the input string.
6 prefix_hash = 0 # Hash value of the prefix.
7 suffix_hash = 0 # Hash value of the suffix.
8 multiplicator = 1 # Multiplicator value used for hash computation.
9 longest_palindrome_idx = 0 # End index of the longest palindromic prefix.
10
11 # Compute rolling hash from both ends.
12 for i, ch in enumerate(s):
13 # Update the prefix hash by appending character.
14 prefix_hash = (prefix_hash * base + (ord(ch) - ord('a') + 1)) % mod
15
16 # Update the suffix hash by adding character (considered at the beginning).
17 suffix_hash = (suffix_hash + (ord(ch) - ord('a') + 1) * multiplicator) % mod
18
19 # Update the multiplicator for the next character.
20 multiplicator = (multiplicator * base) % mod
21
22 # If the prefix and suffix hashes match, update the longest prefix palindrome index.
23 if prefix_hash == suffix_hash:
24 longest_palindrome_idx = i + 1
25
26 # If the entire string is a palindrome, return it.
27 if longest_palindrome_idx == n:
28 return s
29
30 # Otherwise, append the reverse of the remaining suffix to the front to make the shortest palindrome.
31 return s[longest_palindrome_idx:][::-1] + s
32
1public class Solution {
2
3 // This method finds the shortest palindrome starting from the first character by appending characters to the front
4 public String shortestPalindrome(String s) {
5 // we use a prime number as a base for computing rolling hash
6 final int base = 131;
7 // modular multiplication factor, initially 1
8 int multiplier = 1;
9 // we will use a large prime number to mod the result to avoid overflow
10 final int mod = (int) 1e9 + 7;
11 // rolling hash from the front
12 int prefixHash = 0;
13 // rolling hash from the back
14 int suffixHash = 0;
15 // the index till the string is a palindrome
16 int palindromeIdx = 0;
17 // length of the string
18 int length = s.length();
19
20 // iterate through the string to update the prefix and suffix hashes
21 for (int i = 0; i < length; ++i) {
22 // convert character to number (assuming lowercase 'a' to 'z')
23 int charValue = s.charAt(i) - 'a' + 1;
24 // update the prefix hash and ensure it is within the bounds by taking modulo
25 prefixHash = (int) (((long) prefixHash * base + charValue) % mod);
26 // update the suffix hash and ensure it is within the bounds by taking modulo
27 suffixHash = (int) ((suffixHash + (long) charValue * multiplier) % mod);
28 // update the multiplier for the next character
29 multiplier = (int) (((long) multiplier * base) % mod);
30
31 // if the prefix and suffix are equal, then we know the string up to index i is a palindrome
32 if (prefixHash == suffixHash) {
33 palindromeIdx = i + 1;
34 }
35 }
36
37 // If the whole string is a palindrome, return it as is
38 if (palindromeIdx == length) {
39 return s;
40 }
41
42 // We need to add the reverse of the substring from palindromeIdx to the end to the front
43 // to make the string a palindrome
44 String suffixToBeAdded = new StringBuilder(s.substring(palindromeIdx)).reverse().toString();
45
46 // Return the string with the suffix added in front to form the shortest palindrome
47 return suffixToBeAdded + s;
48 }
49}
50
1typedef unsigned long long ull;
2
3class Solution {
4public:
5 string shortestPalindrome(string s) {
6 // Define constants and initial values
7 const int kBase = 131; // Base for polynomial hashing
8 ull prefixHash = 0; // Hash value for the prefix
9 ull suffixHash = 0; // Hash value for the suffix
10 ull currentMultiplier = 1; // Used to compute hash values
11 int palindromeEndIndex = 0; // Index marking the end of the longest palindrome starting at position 0
12 int n = s.size(); // Size of the input string
13
14 // Loop through the string character by character
15 for (int i = 0; i < n; ++i) {
16 int charValue = s[i] - 'a' + 1; // Convert char to int (1-based for 'a' to 'z')
17 prefixHash = prefixHash * kBase + charValue; // Update prefix hash polynomially
18 suffixHash = suffixHash + currentMultiplier * charValue; // Update suffix hash
19 currentMultiplier *= kBase; // Update the base multiplier for the next character
20
21 // If the current prefix is a palindrome (checked by comparing its hash with the suffix hash)
22 if (prefixHash == suffixHash) {
23 palindromeEndIndex = i + 1; // Update the end index of the longest palindrome found
24 }
25 }
26
27 // If the whole string is a palindrome, return it as is
28 if (palindromeEndIndex == n) return s;
29
30 // Otherwise, construct the shortest palindrome by appending the reverse of the remaining substring
31 string remainingSubstring = s.substr(palindromeEndIndex, n - palindromeEndIndex);
32 reverse(remainingSubstring.begin(), remainingSubstring.end());
33 return remainingSubstring + s; // Concatenate the reversed substring with the original string
34 }
35};
36
1// Define a type for unsigned long long equivalent in TypeScript
2type ULL = bigint;
3
4// Converts a lowercase character to an integer (1-based)
5const charToInt = (char: string): number => char.charCodeAt(0) - 'a'.charCodeAt(0) + 1;
6
7// Reverses a string in place
8const reverseString = (s: string): string => s.split('').reverse().join('');
9
10// Computes the shortest palindrome that can be formed by adding characters in front of the given string
11const shortestPalindrome = (s: string): string => {
12 // Constants and initial values
13 const base: ULL = BigInt(131); // Base for polynomial hashing
14 let prefixHash: ULL = BigInt(0); // Hash value for the prefix
15 let suffixHash: ULL = BigInt(0); // Hash value for the suffix
16 let currentMultiplier: ULL = BigInt(1); // Used to compute hash values
17 let palindromeEndIndex = 0; // Index marking the end of the longest palindrome at start
18 const n = s.length; // Size of the input string
19
20 // Loop through the string character by character
21 for (let i = 0; i < n; ++i) {
22 const charValue = charToInt(s[i]); // Convert char to int
23 // Update prefix hash polynomially and suffix hash
24 prefixHash = prefixHash * base + BigInt(charValue);
25 suffixHash = suffixHash + currentMultiplier * BigInt(charValue);
26 // Update the base multiplier for hash computation
27 currentMultiplier *= base;
28
29 // If the current prefix is a palindrome (checked by comparing hashes)
30 if (prefixHash === suffixHash) {
31 palindromeEndIndex = i + 1; // Update the end index of the longest palindrome found
32 }
33 }
34
35 // If the whole string is a palindrome, return it
36 if (palindromeEndIndex === n) return s;
37
38 // Construct the shortest palindrome by appending the reversed suffix to the original string
39 const remainingSubstring = s.substring(palindromeEndIndex);
40 return reverseString(remainingSubstring) + s;
41};
42
43// Example usage
44// const result = shortestPalindrome("example");
45// console.log(result);
46
Time and Space Complexity
The given Python code implements an algorithm for finding the shortest palindrome by appending characters to the beginning of the given string s
. The algorithm is based on calculating hash values from both ends (prefix and suffix) and checking for palindromes.
Time Complexity
The time complexity of this code primarily comes from a single for loop that iterates through each character in the string once. Inside the loop, it computes the prefix and suffix hash values, and compares them to check if they are equal.
Here's the breakdown:
- The hash operations and comparison inside the for loop are O(1) operations as they are done using arithmetic calculations.
- The for loop runs n times, where n is the length of the string
s
.
Therefore, the time complexity of this code is O(n)
, where n
is the length of the input string.
Space Complexity
The space complexity of the code is determined by the storage used which is independent of the length of the input string s
.
Here's the breakdown:
- Variables
prefix
,suffix
,mul
, andidx
are integers which occupy constant space. - The slice and reverse operation
s[idx:][::-1]
creates a new string of at mostn-1
characters when the input string is not already a palindrome.
Even though a new string is created in the worst-case scenario, the space complexity is proportional to the input string size which gives us O(n)
. However, if we consider only the additional space excluding the input and output, the space complexity is actually O(1)
since we're only using a fixed amount of additional storage regardless of the input size.
Learn more about how to find time and space complexity quickly using problem constraints.
Which of the tree traversal order can be used to obtain elements in a binary search tree in sorted order?
Recommended Readings
LeetCode Patterns Your Personal Dijkstra's Algorithm to Landing Your Dream Job The goal of AlgoMonster is to help you get a job in the shortest amount of time possible in a data driven way We compiled datasets of tech interview problems and broke them down by patterns This way we
Recursion Recursion is one of the most important concepts in computer science Simply speaking recursion is the process of a function calling itself Using a real life analogy imagine a scenario where you invite your friends to lunch https algomonster s3 us east 2 amazonaws com recursion jpg You first
Runtime Overview When learning about algorithms and data structures you'll frequently encounter the term time complexity This concept is fundamental in computer science and offers insights into how long an algorithm takes to complete given a certain input size What is Time Complexity Time complexity represents the amount of time
Want a Structured Path to Master System Design Too? Don’t Miss This!