Rearrange Spaces Between Words

You are given a string text of words that are placed among some number of spaces. Each word consists of one or more lowercase English letters and are separated by at least one space. It's guaranteed that text contains at least one word.

Rearrange the spaces so that there is an equal number of spaces between every pair of adjacent words and that number is maximized. If you cannot redistribute all the spaces equally, place the extra spaces at the end, meaning the returned string should be the same length as text.

Return the string after rearranging the spaces.

LeetCode Problem - 1592

class Solution {
    public String reorderSpaces(String text) {
        // Step 1: Count the number of spaces in the input text
        int count = 0;
        for (int i = 0; i < text.length(); i++) {
            if (text.charAt(i) == ' ') {
                count++;
            }
        }

        // Step 2: Split the text into words, removing extra spaces
        String[] newStr = text.trim().replaceAll("\\s+", " ").split(" ");
        int length = newStr.length; // number of words

        // Step 3: Handle the edge case where there's only one word
        if (length == 1) {
            // If there's only one word, append all spaces at the end of the word
            return newStr[0] + " ".repeat(count);
        }

        // Step 4: Calculate the number of spaces to distribute between words
        int space = count / (length - 1); // spaces between words
        int remainingSpaces = count % (length - 1); // leftover spaces after even distribution

        // Step 5: Rebuild the string with evenly distributed spaces
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < length; i++) {
            sb.append(newStr[i]); // append the word
            if (i < length - 1) {
                // Add spaces between words, except after the last word
                sb.append(" ".repeat(space));
            }
        }

        // Step 6: Append any remaining spaces at the end of the sentence
        sb.append(" ".repeat(remainingSpaces));

        // Return the final result
        return sb.toString();
    }
}