如何有效地查找首尾相同的字符串中子字符串的数量？

Question

我正在尝试查找具有相同的第一个和最后一个字符的字符串中子字符串的数量。

我可以通过两个for循环以幼稚的方式解决它。

我觉得可以更有效地解决它。

如何以更有效的方式解决它？

Answer 1

遍历字符串，并计算每个不同字符的出现次数。 然后，如果一个字符仅出现一次，则没有以该字符开头和结尾的子字符串，如果出现两次，则只有1次，如果出现3次，则只有3次，如果发生4次，则是6次。函数是C（ n，2）= n！/（2！（n-2）！）= n（n-1）/ 2。

这是一个潜在的实现。

inline int Nchoose2(int n) {
  return n*(n-1)/2;
}

std::string s;
std::map<char,int> m;
int cnt = 0;

for (char c : s) {
  if (!m.count(c)) m[c] = 0;
  else ++m[c];
}

for (auto &c : m)
   cnt += Nchoose2(c.second);