Leetcode problem 55 - Truncate Sentence

A sentence is a list of words that are separated by a single space with no leading or trailing spaces. Each of the words consists of only uppercase and lowercase English letters (no punctuation).

For example, “Hello World”, “HELLO”, and “hello world hello world” are all sentences. You are given a sentence s​​​​​​ and an integer k​​​​​​. You want to truncate s​​​​​​ such that it contains only the first k​​​​​​ words. Return s​​​​​​ after truncating it.

Example 1:

Input: s = “Hello how are you Contestant”, k = 4 Output: “Hello how are you” Explanation: The words in s are [“Hello”, “how” “are”, “you”, “Contestant”]. The first 4 words are [“Hello”, “how”, “are”, “you”]. Hence, you should return “Hello how are you”. Example 2:

Input: s = “What is the solution to this problem”, k = 4 Output: “What is the solution” Explanation: The words in s are [“What”, “is” “the”, “solution”, “to”, “this”, “problem”]. The first 4 words are [“What”, “is”, “the”, “solution”]. Hence, you should return “What is the solution”. Example 3:

Input: s = “chopper is not a tanuki”, k = 5 Output: “chopper is not a tanuki”

Constraints:

1 <= s.length <= 500 k is in the range [1, the number of words in s]. s consist of only lowercase and uppercase English letters and spaces. The words in s are separated by a single space. There are no leading or trailing spaces.

/**
 * @param {string} s
 * @param {number} k
 * @return {string}
 */
var truncateSentence = function (s, k) {
  let words = s.split(" ");
  let result = [];

  for (let i = 0; i < k; i++) {
    if (i === k - 1) {
      result.push(words[i]);
    } else {
      result.push(words[i] + " ");
    }
  }
  return result.join("");
};

I have s and k value. s is a string words, and k is a integer value. I need to split s words into array, and get rid of elements in the array after index k. So, I’m going to split s words first, and then, initialize an empty array to store elements before index k.

To put space between each words, I used if conditional statement. If i is reached at the last of element in the array, last element is pushed without space at the end of the element.

I used join method to make the array string, and return it.