Question

1704. 判断字符串的两半是否相似

English Version

题目描述

给你一个偶数长度的字符串 s 。将其拆分成长度相同的两半，前一半为 a ，后一半为 b 。

两个字符串 相似 的前提是它们都含有相同数目的元音（'a'，'e'，'i'，'o'，'u'，'A'，'E'，'I'，'O'，'U'）。注意，s 可能同时含有大写和小写字母。

如果_ _a_ _和_ _b 相似，返回 true ；否则，返回 false 。

 

示例 1：

输入：s = "book"
输出：true
解释：a = "bo" 且 b = "ok" 。a 中有 1 个元音，b 也有 1 个元音。所以，a 和 b 相似。

示例 2：

输入：s = "textbook"
输出：false
解释：a = "text" 且 b = "book" 。a 中有 1 个元音，b 中有 2 个元音。因此，a 和 b 不相似。
注意，元音 o 在 b 中出现两次，记为 2 个。

 

提示：

• 2 <= s.length <= 1000
• s.length 是偶数
• s 由 大写和小写 字母组成

解法

方法一：计数

遍历字符串，若字符串前半段的元音个数等于后半段的元音个数，则返回 true，否则返回 false。

时间复杂度 $O(n)$，其中 $n$ 为字符串的长度。空间复杂度 $O(C)$，其中 $C$ 为元音字母的个数。

class Solution:
  def halvesAreAlike(self, s: str) -> bool:
    cnt, n = 0, len(s) >> 1
    vowels = set('aeiouAEIOU')
    for i in range(n):
      cnt += s[i] in vowels
      cnt -= s[i + n] in vowels
    return cnt == 0

class Solution {
  private static final Set<Character> VOWELS
    = Set.of('a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U');

  public boolean halvesAreAlike(String s) {
    int cnt = 0, n = s.length() >> 1;
    for (int i = 0; i < n; ++i) {
      cnt += VOWELS.contains(s.charAt(i)) ? 1 : 0;
      cnt -= VOWELS.contains(s.charAt(i + n)) ? 1 : 0;
    }
    return cnt == 0;
  }
}

class Solution {
public:
  bool halvesAreAlike(string s) {
    unordered_set<char> vowels = {'a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U'};
    int cnt = 0, n = s.size() / 2;
    for (int i = 0; i < n; ++i) {
      cnt += vowels.count(s[i]);
      cnt -= vowels.count(s[i + n]);
    }
    return cnt == 0;
  }
};

func halvesAreAlike(s string) bool {
  vowels := map[byte]bool{}
  for _, c := range "aeiouAEIOU" {
    vowels[byte(c)] = true
  }
  cnt, n := 0, len(s)>>1
  for i := 0; i < n; i++ {
    if vowels[s[i]] {
      cnt++
    }
    if vowels[s[i+n]] {
      cnt--
    }
  }
  return cnt == 0
}

function halvesAreAlike(s: string): boolean {
  const set = new Set(['a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U']);
  const n = s.length >> 1;
  let count = 0;
  for (let i = 0; i < n; i++) {
    set.has(s[i]) && count++;
    set.has(s[n + i]) && count--;
  }
  return count === 0;
}

use std::collections::HashSet;
impl Solution {
  pub fn halves_are_alike(s: String) -> bool {
    let set: HashSet<&u8> = [b'a', b'e', b'i', b'o', b'u', b'A', b'E', b'I', b'O', b'U']
      .into_iter()
      .collect();
    let s = s.as_bytes();
    let n = s.len() >> 1;
    let mut count = 0;
    for i in 0..n {
      if set.contains(&s[i]) {
        count += 1;
      }
      if set.contains(&s[n + i]) {
        count -= 1;
      }
    }
    count == 0
  }
}

/**
 * @param {string} s
 * @return {boolean}
 */
var halvesAreAlike = function (s) {
  const str = 'aeiouAEIOU';
  let cnt = 0;
  for (let i = 0; i < s.length / 2; i++) {
    if (str.indexOf(s[i]) > -1) cnt++;
    if (str.indexOf(s[s.length - 1 - i]) > -1) cnt--;
  }
  return cnt === 0;
};

class Solution {
  /**
   * @param String $s
   * @return Boolean
   */
  function halvesAreAlike($s) {
    $cnt = 0;
    for ($i = 0; $i < strlen($s) / 2; $i++) {
      if (strpos('aeiouAEIOU', $s[$i]) !== false) {
        $cnt++;
      }
      if (strpos('aeiouAEIOU', $s[strlen($s) / 2 + $i]) !== false) {
        $cnt--;
      }
    }
    return $cnt == 0;
  }
}

方法二

class Solution:
  def halvesAreAlike(self, s: str) -> bool:
    vowels = set('aeiouAEIOU')
    a, b = s[: len(s) >> 1], s[len(s) >> 1 :]
    return sum(c in vowels for c in a) == sum(c in vowels for c in b)