Codeforces 451D Count Good Substrings(組合數學)，codeforces451d

Codeforces 451D Count Good Substrings(組合數學)，codeforces451d

題目連結：Codeforces 451D Count Good Substrings

題目大意：定義good string，就是就一個字串的連續相同字元用一個該字元替代後，形成迴文串的字串。現在給出一個字串，問說該字串的子串中，為good string的串有多少個，分長度為奇數和偶數的輸出。

解題思路：因為字串的組成為a和b，所以只要是頭尾相同的子串都是滿足的。所以我們計算在奇數和偶數位置的奇數個數和偶數個數即可，然後用組合數學求出答案。

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;typedef long long ll;const int maxn = 1e5+5;char str[maxn];ll cal (ll u) {    return u * (u-1) / 2;}int main () {    ll oddA = 0, evenA = 0;    ll oddB = 0, evenB = 0;    scanf("%s", str);    int len = strlen(str);    for (int i = 0; i < len; i++) {        if (str[i] == 'a') {            if (i&1)                oddA++;            else                evenA++;        } else {            if (i&1)                oddB++;            else                evenB++;        }    }    ll even = oddA * evenA + oddB * evenB;    ll odd = cal(oddA) + cal(evenA) + cal(oddB) + cal(evenB) + len;    printf("%lld %lld\n", even, odd);    return 0;}