环绕字符串中唯一的子字符串

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第32天，点击查看活动详情

467. 环绕字符串中唯一的子字符串 - 力扣（Leetcode）

把字符串 s 看作是 “abcdefghijklmnopqrstuvwxyz” 的无限环绕字符串，所以 s 看起来是这样的：

• "...zabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcd...." . 

现在给定另一个字符串 p 。返回 s 中 唯一 的 p 的 非空子串 的数量 。 

示例 1:

输入: p = "a"
输出: 1
解释: 字符串 s 中只有一个"a"子字符。

示例 2:

输入: p = "cac"
输出: 2
解释: 字符串 s 中的字符串“cac”只有两个子串“a”、“c”。.

示例 3:

输入: p = "zab"
输出: 6
解释: 在字符串 s 中有六个子串“z”、“a”、“b”、“za”、“ab”、“zab”。

提示:

• 1 <= p.length <= 10^5
• p 由小写英文字母构成

思路

由题目知s中的前后字符都是连续的，s中的非空子串也必然是连续的比如efg、hij、zab等。现在要返回s中唯一的p的非空子串数量，有两个要求：

1. 子串中前后字符要连续，
2. 子串不能重复。

我们假设一个字符p[j]在连续字符串p中位置为i(i从1开始计数)，那以这个字符结尾的连续子串有i个，分别为p[j]、p[j-1]p[j]、p[j-2]p[j-1]p[j]、...、p[j-i+1]p[j-i+2]...p[j-1]p[j]，当同一个字符x在两个连续子串p1和p2中的位置分别为m和n，且m < n时，p2中以x结尾的连续子串包含p1中以x结尾的连续子串。

解本题时，我们可以遍历p，求出p[i]在连续子串中的位置，当同一字符出现在不同的连续子串中时，取最大值。最后求所有字符位置和就是连续子字符的个数。

解题

/**
 * @param {string} p
 * @return {number}
 */
var findSubstringInWraproundString = function (p) {
  const a = "a".charCodeAt(0);
  const lens = new Array(26).fill(0);
  let k = 1;
  let prev = -2;
  for (let i = 0; i < p.length; i++) {
    const index = p.charCodeAt(i) - a;
    const diff = index - prev;
    if (diff === 1 || diff === -25) {
      k++;
    } else {
      k = 1;
    }
    prev = index
    lens[index] = Math.max(k, lens[index]);
  }
  return lens.reduce((sum, num) => (sum + num));
};