代码源每日一题DIV2 #563. 子串分值和

代码源每日一题DIV2 #563. 子串分值和

• 思路
  • ans = 出现a的所有子串数量+出现b的所有子串数量+…+出现z的所有子串数量
  • f（aaa）为什么=1，因为它只属于出现a的子串，f（abc）为什么=3，因为它属于出现a的子串也属于出现b的子串也属于出现c的子串
• 学到的东西
  • 计数问题—》枚举可能出现的字符而非枚举每一个串
  • 正难则反，要找到含有a的所有子串很难，那么就用（所有子串数量-不含a的子串数量）
  • 对于一个长度为len的字符串，它的所有子串数量为 $（len\ast （len+1）/2）$
• AC代码

void solve() {
    map<char,vector<int>> m;
    string s;cin>>s;
    s='0'+s;
    for (int i = 1; i < s.size(); ++i) {
        if (m[s[i]].empty()) m[s[i]].push_back(0);
        m[s[i]].push_back(i);
    }
    int ans = 0;
    int otot = s.size()*(s.size()-1)/2;
    for (char i = 'a'; i <='z' ; ++i) {
        if (m[i].empty()) continue;
        int tot = 0;
        vector<int> temp = m[i];
        temp.push_back(s.size());
        for (int j = 0; j < temp.size()-1; ++j) {
            int len = temp[j+1]-temp[j]-1;
            tot += (len*(len+1)/2);
        }
        ans+=(otot-tot);
    }
    cout<<ans<<endl;
}