题目链接:传送门
很容易想到记录每个字幕出现的位置,按照每个字母出现的位置来计算对答案产生的贡献
如果说当前位置为posj,前一次出现了的位置为posi,下一次出现的位置为posk,那么当前位置posj对答案产生的贡献是:
(posj - posi) * (posk - posj)
code:
class Solution {
public:
int uniqueLetterString(string s) {
int ans = 0;
unordered_map<int,vector<int> > mp;
for(int i = 0;i < s.size(); i ++) {
if(mp[s[i]].size() == 0) {
mp[s[i]].push_back(-1);
}
mp[s[i]].push_back(i);
}
unordered_map<int, vector<int> >::iterator it;
for(it = mp.begin(); it != mp.end(); it ++) {
char c = it->first;
vector<int> v = it->second;
v.push_back(s.size());
for(int i=1;i<v.size() - 1;i ++) {
ans += (v[i] - v[i-1]) * (v[i+1] - v[i]);
}
}
return ans;
}
};