1790. Check if One String Swap Can Make Strings Equal

EasyHash TableStringCounting
Leetcode Link

Problem Description

Given two strings s1 and s2 of equal length, our task is to determine if we can make them equal by performing at most one string swap on exactly one of the strings. A string swap involves choosing any two indices within one string and swapping the characters at these indices. We are to return true if the strings can be equalized in this manner or false if it's not possible.

Intuition

To solve this problem, we must consider that if two strings are to be equal after at most one swap, then there can only be two characters that differ between them. If there are no characters that differ, the strings are already equal. If there is only one pair of characters that differ, swapping them in either of the strings would make the strings equal. However, if there are more than two characters that differ, it will not be possible to make the strings equal with just one swap.

To implement this intuition in code, we iterate through both strings simultaneously, comparing corresponding characters. We keep a count of how many characters differ (cnt) and remember the characters that differed on the first occurrence (c1 and c2). If we encounter a second occurrence of differing characters, we check whether they can form a valid swap with the first occurrence. If they can't, or if we find a third occurrence, we immediately know it's impossible to equalize the strings with one swap and return false. If the function has not returned false, we check the total count of differing characters. We should return true only if the count is exactly 0 or 2, which means the strings were initially equal, or they can be equalized with one swap. A count of 1 indicates there's a single difference, which can't be resolved with a swap, hence we return false.

Not Sure What to Study? Take the 2-min Quiz to Find Your Missing Piece:

Which algorithm should you use to find a node that is close to the root of the tree?

Solution Approach

The solution approach employs a simple iteration wherein we traverse both strings s1 and s2 using a single for loop. Throughout the iteration, we keep a count (cnt) of the positions where the characters in s1 and s2 are different. Additionally, we make use of two variables c1 and c2 to hold the pair of characters that were different on the first occurrence.

In Python, the zip function is utilized to iterate over characters from both strings simultaneously, providing a convenient way to compare characters at the same indices from s1 and s2. When a difference is found (i.e., characters a from s1 and b from s2 are not the same), the cnt counter is incremented.

The algorithm considers three main scenarios:

  1. If after the traversal, the value of cnt is zero, it means that all the characters at corresponding indices are the same, and hence, s1 is already equal to s2. Thus we can return True.

  2. If cnt becomes greater than 2 at any point during the iteration, it means there are too many differences to correct with a single swap, and the function immediately returns False.

  3. Lastly, if the function encounters exactly two differences (cnt equals 2), it checks if the current pair of differing characters (a, b) could be swapped with the first pair (c1, c2) to make the strings equal. This is assessed by checking if the current character from s1 (a) equals the first differing character from s2 (c2), and the current character from s2 (b) equals the first differing character from s1 (c1). If this condition doesn't hold, then it's not possible with a single swap to make the strings equal, and we return False.

At the end of the iteration, we also need to return False if cnt is exactly one, because a single difference can't be rectified with a swap. Hence, we can conclude that True is only returned if cnt is exactly 0 or 2, signifying that no swaps or exactly one swap can equalize the strings.

Discover Your Strengths and Weaknesses: Take Our 2-Minute Quiz to Tailor Your Study Plan:

Which data structure is used in a depth first search?

Example Walkthrough

Let's consider a small example with the strings s1 = "converse" and s2 = "conserve". As per the description, we need to determine if a single swap can make these two strings equal.

  1. We start iterating through both strings from the first character: s1[0] = 'c' and s2[0] = 'c'. Since the characters are the same, we move on.

  2. The same goes for the second, third, and fourth characters: s1[1] = 'o' and s2[1] = 'o', s1[2] = 'n' and s2[2] = 'n', s1[3] = 'v' and s2[3] = 'v'. All are equal; hence cnt remains 0.

  3. When we reach the fifth character, we notice a difference: s1[4] = 'e' and s2[4] = 's'. This is our first difference, so cnt is incremented to 1 and we store the differing characters in variables: c1 = 'e' and c2 = 's'.

  4. The sixth characters are equal again: s1[5] = 'r' and s2[5] = 'r'.

  5. At the seventh character, there's another difference: s1[6] = 's' and s2[6] = 'e'. We increment cnt to 2 and note this second pair of differing characters.

  6. Since cnt equals 2, we now check if the characters s1[6] and s2[4] would form a valid swap with the first pair, c1 and c2. We find that s1[6] equals c2, and s2[6] equals c1, meaning swapping s1[4] with s1[6] will make s1 equal to s2.

  7. The remaining characters s1[7] = 'e' and s2[7] = 'r', s1[8] = 'r' and s2[8] = 'v' are also equal, confirming that cnt remains 2 by the end of the iteration and no further discrepancies arise.

  8. Because cnt is exactly 2 and the single swap condition is met, we return True. The single swap needed would be to swap s1[4] with s1[6] resulting in s1 becoming conserve, which is equal to s2.

Using this approach allows us to go through each character of both strings and efficiently determine whether a single swap can make the two strings equal without any unnecessary comparisons or operations.

Solution Implementation

1class Solution:
2    def areAlmostEqual(self, string1: str, string2: str) -> bool:
3        # Initialize the count of different characters and placeholders for characters that differ.
4        difference_count = 0
5        char1 = char2 = None
6      
7        # Iterate through characters of both strings in parallel.
8        for char_string1, char_string2 in zip(string1, string2):
9            # If characters don't match, increase the difference count.
10            if char_string1 != char_string2:
11                difference_count += 1
12                # Check if there are more than 2 differences or if the swap doesn't make strings equal
13                if difference_count > 2 or (difference_count == 2 and (char_string1 != char2 or char_string2 != char1)):
14                    return False
15                # Record the first set of different characters.
16                char1, char2 = char_string1, char_string2
17
18        # If there's exactly one difference, the strings cannot be made equal with one swap.
19        return difference_count != 1
20
1class Solution {
2    public boolean areAlmostEqual(String s1, String s2) {
3        // Initialize a counter for the number of non-matching character pairs between s1 and s2.
4        int mismatchCount = 0; 
5      
6        // Initialize variables to store characters from non-matching character pairs.
7        char firstCharFromS1 = 0, firstCharFromS2 = 0; 
8      
9        // Iterate over the strings to compare characters at each index.
10        for (int i = 0; i < s1.length(); ++i) {
11            // Get the current characters from each string.
12            char charFromS1 = s1.charAt(i), charFromS2 = s2.charAt(i); 
13          
14            // Check for non-matching characters
15            if (charFromS1 != charFromS2) {
16              
17                // If more than two non-matching pairs, strings are already not almost equal.
18                if (++mismatchCount > 2 || 
19                    // If this is the second mismatch but does not form a transposable pair with the first mismatch, return false.
20                    (mismatchCount == 2 && !(charFromS1 == firstCharFromS2 && charFromS2 == firstCharFromS1))) {
21                    return false;
22                }
23                // Store the characters from the first non-matching character pair.
24                firstCharFromS1 = charFromS1;
25                firstCharFromS2 = charFromS2;
26            }
27        }
28        // If there is exactly one mismatch, strings cannot be made equal by a single swap.
29        // Strings are almost equal if there were zero or two mismatches.
30        return mismatchCount != 1;
31    }
32}
33
1class Solution {
2public:
3    bool areAlmostEqual(string str1, string str2) {
4        // Initialize a counter to track the number of positions where str1 and str2 differ
5        int differenceCount = 0;
6        // Variables to store the characters from each string when a mismatch is found
7        char charFromStr1 = 0, charFromStr2 = 0;
8      
9        // Iterate through both strings to compare character by character
10        for (int index = 0; index < str1.size(); ++index) {
11            char charA = str1[index], charB = str2[index];
12          
13            // If there is a mismatch, we'll need to check further
14            if (charA != charB) {
15                // Increment the difference counter and check if it exceeds 2. If it does, return false as more than one swap won't make them equal
16                if (++differenceCount > 2 || (differenceCount == 2 && (charA != charFromStr2 || charB != charFromStr1))) {
17                    return false;
18                }
19                // Update the characters that were found to mismatch for comparison when the next mismatch occurs
20                charFromStr1 = charA, charFromStr2 = charB;
21            }
22        }
23        // If there was exactly one mismatch, strings cannot be made equal with a single swap
24        // Return true if difference count is 0 or 2 (since they can be equal after exactly one swap); otherwise, return false
25        return differenceCount != 1;
26    }
27};
28
1// This function checks if two strings are almost equal by allowing one swap of two characters in one string
2
3function areAlmostEqual(string1: string, string2: string): boolean {
4    let firstMismatchedCharFromS1: string; // to store the character from string1 involved in the first mismatch
5    let firstMismatchedCharFromS2: string; // to store the character from string2 involved in the first mismatch
6    let mismatchCount = 0; // to keep track of the number of mismatches found
7
8    // Loop over each character in the strings to check for mismatches
9    for (let i = 0; i < string1.length; ++i) {
10        const charFromS1 = string1.charAt(i);
11        const charFromS2 = string2.charAt(i);
12
13        // If a mismatch is found
14        if (charFromS1 !== charFromS2) {
15            mismatchCount++; // we increment the mismatch counter
16
17            // If more than two mismatches are found, or if at the second mismatch the
18            // mismatching characters are not the transposed characters from the first mismatch,
19            // then the strings cannot be made equal with one swap.
20            if (mismatchCount > 2 || (mismatchCount === 2 && (charFromS1 !== firstMismatchedCharFromS2 || charFromS2 !== firstMismatchedCharFromS1))) {
21                return false;
22            }
23
24            // If this is the first mismatch encountered, store the mismatching characters
25            if (mismatchCount === 1) {
26                firstMismatchedCharFromS1 = charFromS1;
27                firstMismatchedCharFromS2 = charFromS2;
28            }
29        }
30    }
31
32    // Strings are considered almost equal if there are no mismatches or exactly two mismatches
33    return mismatchCount !== 1;
34}
35
Not Sure What to Study? Take the 2-min Quiz:

Which algorithm should you use to find a node that is close to the root of the tree?

Time and Space Complexity

The code provided implements a function to check if two strings are almost equal. That means they are equal or they can become equal by swapping at most one pair of characters in one of the strings.

Time Complexity:

The time complexity of the given code is O(n), where n is the length of the strings s1 and s2. This time complexity arises because the code iterates over each character of the strings exactly once through the use of the zip function.

The conditional statement inside the loop has constant-time complexity checks (O(1)), thus, they don't affect the overall linear time complexity of iterating through the strings.

Space Complexity:

The space complexity of the given code is O(1). No additional space that scales with the input size is required. The variables cnt, c1, and c2 use constant space, only storing a fixed number of elements (at most two characters and a counter) regardless of the input size.

Since the code operates in-place, checking the characters of the input strings without creating any additional data structures or recursive calls, the space complexity remains constant.

Learn more about how to find time and space complexity quickly using problem constraints.

Fast Track Your Learning with Our Quick Skills Quiz:

Problem: Given a list of tasks and a list of requirements, compute a sequence of tasks that can be performed, such that we complete every task once while satisfying all the requirements.

Which of the following method should we use to solve this problem?


Recommended Readings


Got a question? Ask the Teaching Assistant anything you don't understand.

Still not clear? Ask in the Forum,  Discord or Submit the part you don't understand to our editors.


TA 👨‍🏫