1816. 截断句子

1816. 截断句子

难度 简单
句子 是一个单词列表，列表中的单词之间用单个空格隔开，且不存在前导或尾随空格。每个单词仅由大小写英文字母组成（不含标点符号）。

  • 例如，​​"Hello World"​​​、​​"HELLO"​​​ 和​​"hello world hello world"​​​ 都是句子。
    给你一个句子​​​s​​​​​​​ 和一个整数​​k​​​​​​​ ，请你将​​s​​​ 截断 ​，​​​使截断后的句子仅含 前​​k​​​​​​​ 个单词。返回 截断​​s​​​​​​​ 后得到的句子。

示例 1：

输入：s = "Hello how are you Contestant", k = 4
输出："Hello how are you"
解释：
s 中的单词为 ["Hello", "how" "are", "you", "Contestant"]
前 4 个单词为 ["Hello", "how", "are", "you"]
因此，应当返回 "Hello how are you"

示例 2：

输入：s = "What is the solution to this problem", k = 4
输出："What is the solution"
解释：
s 中的单词为 ["What", "is" "the", "solution", "to", "this", "problem"]
前 4 个单词为 ["What", "is", "the", "solution"]
因此，应当返回 "What is the solution"

示例 3：

输入：s = "chopper is not a tanuki", k = 5
输出："chopper is not a tanuki"

提示：

  • 1 <= s.length <= 500
  • k 的取值范围是 [1, s 中单词的数目]
  • s 仅由大小写英文字母和空格组成
  • s 中的单词之间由单个空格隔开
  • 不存在前导或尾随空格

题解

class Solution {
    public String truncateSentence(String s, int k) {
        String [] strArray = s.split(" ");
        String ans="";
        int first=1;
        for(int i=0;i<k;i++){
            if(first==1){
                ans+=strArray[i];
                first=0;
            }else{
                ans+=" "+strArray[i];
            }
             
        }
        return ans;
    }
}

1816. 截断句子【我亦无他唯手熟尔】_i++

官方

方法一：遍历
思路与算法

由题意可知，除了最后一个单词，每个单词后面都跟随一个空格。因此我们可以通过统计空格与句子结尾的数目来统计单词数 count。当 count=k 时，将当前的下标记录到end，返回句子s 在end 处截断的句子。

代码

class Solution {
    public String truncateSentence(String s, int k) {
        int n = s.length();
        int end = 0, count = 0;
        for (int i = 1; i <= n; i++) {
            if (i == n || s.charAt(i) == ' ') {
                count++;
                if (count == k) {
                    end = i;
                    break;
                }
            }
        }
        return s.substring(0, end);
    }
}

复杂度分析

  • 时间复杂度：O(N)，其中 NN 为句子 \textit{s}s 的长度。遍历整个字符串需要 O(N)。
  • 空间复杂度：O(1)。注意返回值不计入空间复杂度。

1816. 截断句子【我亦无他唯手熟尔】_i++_02