描述
我们定义了一个函数 countUniqueChars(s) 来统计字符串 s 中的唯一字符,并返回唯一字符的个数。
例如:s = “LEETCODE” ,则其中 “L”, “T”,“C”,“O”,“D” 都是唯一字符,因为它们只出现一次,所以 countUniqueChars(s) = 5 。
本题将会给你一个字符串 s ,我们需要返回 countUniqueChars(t) 的总和,其中 t 是 s 的子字符串。输入用例保证返回值为 32 位整数。
注意,某些子字符串可能是重复的,但你统计时也必须算上这些重复的子字符串(也就是说,你必须统计 s 的所有子字符串中的唯一字符)。
示例 1:
输入: s = "ABC"
输出: 10
解释: 所有可能的子串为:"A","B","C","AB","BC" 和 "ABC"。
其中,每一个子串都由独特字符构成。
所以其长度总和为:1 + 1 + 1 + 2 + 2 + 3 = 10
示例 2:
输入: s = "ABA"
输出: 8
解释: 除了 countUniqueChars("ABA") = 1 之外,其余与示例 1 相同。
示例 3:
输入:s = "LEETCODE"
输出:92
提示:
1 <= s.length <= 10^5
s 只包含大写英文字符
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/count-unique-characters-of-all-substrings-of-a-given-string
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
由计算每一个子字符串的单一字符的个数转变成计算一个字母在是唯一字母的时候能够给周围多少个子字符串充当唯一字母。
- 用数组存储一个字母在整个字符串所有出现过的下标位置
- 用这些出现过的位置构造出每个位置的它是唯一字母的区间。
一个字母的一个唯一区间如:[j,k],i是唯一字母的位置,则包含i的所有子字符串的个数是(i-j)*(k-i)。
(注意j,k是边界,不应该出现在包含i的子字符串中)
理由如下:把[j,k]看成两部分,左半边所有是以i结尾的所有子字符串,右半边是所有不以i为开头的子字符串。
左半边的个数是(i-j),右半边除了显著的k-i-1,还应该包括一个空字符串,所有右半边的个数是k-i。
class Solution {
public int uniqueLetterString(String s) {
Map<Character,List<Integer>> map = new HashMap<>();
int t = 0;
for (char c : s.toCharArray()) {
if (!map.containsKey(c)) {
map.put(c,new ArrayList<>());
}
map.get(c).add(t++);
}
int ans = 0;
int head = -1;
int tail = s.length();
for (Character c : map.keySet()) {
List<Integer> list = map.get(c);
for (int i = 0; i < list.size(); i++) {
tail = i == list.size() - 1 ? s.length() : list.get(i+1);
head = i == 0 ? -1 : list.get(i-1);
ans += (tail-list.get(i)) * (list.get(i) - head);
}
}
return ans;
}
}