Question

1455. 检查单词是否为句中其他单词的前缀

English Version

题目描述

给你一个字符串 sentence 作为句子并指定检索词为 searchWord ，其中句子由若干用 单个空格 分隔的单词组成。请你检查检索词 searchWord 是否为句子 sentence 中任意单词的前缀。

如果 searchWord 是某一个单词的前缀，则返回句子 sentence 中该单词所对应的下标（下标从 1 开始）。如果 searchWord 是多个单词的前缀，则返回匹配的第一个单词的下标（最小下标）。如果 searchWord 不是任何单词的前缀，则返回 -1 。

字符串 s 的 前缀 是 s 的任何前导连续子字符串。

 

示例 1：

输入：sentence = "i love eating burger", searchWord = "burg"
输出：4
解释："burg" 是 "burger" 的前缀，而 "burger" 是句子中第 4 个单词。

示例 2：

输入：sentence = "this problem is an easy problem", searchWord = "pro"
输出：2
解释："pro" 是 "problem" 的前缀，而 "problem" 是句子中第 2 个也是第 6 个单词，但是应该返回最小下标 2 。

示例 3：

输入：sentence = "i am tired", searchWord = "you"
输出：-1
解释："you" 不是句子中任何单词的前缀。

 

提示：

• 1 <= sentence.length <= 100
• 1 <= searchWord.length <= 10
• sentence 由小写英文字母和空格组成。
• searchWord 由小写英文字母组成。

解法

方法一：字符串分割

将 $sentence$ 按空格分割为 $words$，然后遍历 $words$，检查 $words[i]$ 是否是 $searchWord$ 的前缀，是则返回 $i+1$。若遍历结束，所有单词都不满足，返回 $-1$。

时间复杂度 $O(mn)$。其中 $m$ 是 $sentence$ 的长度，而 $n$ 是 $searchWord$ 的长度。

class Solution:
  def isPrefixOfWord(self, sentence: str, searchWord: str) -> int:
    for i, s in enumerate(sentence.split(), 1):
      if s.startswith(searchWord):
        return i
    return -1

class Solution {
  public int isPrefixOfWord(String sentence, String searchWord) {
    String[] words = sentence.split(" ");
    for (int i = 0; i < words.length; ++i) {
      if (words[i].startsWith(searchWord)) {
        return i + 1;
      }
    }
    return -1;
  }
}

class Solution {
public:
  int isPrefixOfWord(string sentence, string searchWord) {
    stringstream ss(sentence);
    string s;
    for (int i = 1; ss >> s; ++i) {
      if (s.find(searchWord) == 0) {
        return i;
      }
    }
    return -1;
  }
};

func isPrefixOfWord(sentence string, searchWord string) int {
  for i, s := range strings.Split(sentence, " ") {
    if strings.HasPrefix(s, searchWord) {
      return i + 1
    }
  }
  return -1
}

function isPrefixOfWord(sentence: string, searchWord: string): number {
  const ss = sentence.split(/\s/);
  const n = ss.length;
  for (let i = 0; i < n; i++) {
    if (ss[i].startsWith(searchWord)) {
      return i + 1;
    }
  }
  return -1;
}

impl Solution {
  pub fn is_prefix_of_word(sentence: String, search_word: String) -> i32 {
    let ss = sentence.split_whitespace().collect::<Vec<&str>>();
    for i in 0..ss.len() {
      if ss[i].starts_with(&search_word) {
        return (i + 1) as i32;
      }
    }
    -1
  }
}

class Solution {
  /**
   * @param String $sentence
   * @param String $searchWord
   * @return Integer
   */
  function isPrefixOfWord($sentence, $searchWord) {
    $arr = explode(' ', $sentence);
    for ($i = 0; $i < count($arr); $i++) {
      if (strpos($arr[$i], $searchWord) === 0) {
        return $i + 1;
      }
    }
    return -1;
  }
}