2022AcWing寒假算法每日一题之1884. COW（附上巧妙的位运算方法）

2022AcWing寒假算法每日一题之1884. COW（附上巧妙的位运算方法）

题目链接：AcWing1884. COW

思路：

1. 从前往后遍历字符串，‘O’会和前面的‘C’组成‘CO’，前面有几个‘C’的个数就有几个‘CO’，即o+=c；

2. 同理，‘CO’与‘W’组成‘COW’组合，前面有多少个‘CO’，就有多少个‘COW’，即w+=o；

3. 可利用位运算进行简单处理（方法二）

具体代码如下：

#include <iostream>
#include <stack>
#include <math.h>
#include <cstring>
#include <algorithm>
using namespace std;
int n;
long long c,o,w;//防止溢出
string s;
int main()
{
    cin>>n;
    cin>>s;
    for(int i=0;i<s.length();i++)
    {
        if(s[i]=='C')c++;//C的个数
        else if(s[i]=='O')o+=c;//CO的个数
        else w+=o;//COW的个数
    }
    cout<<w<<endl;
    return 0;
}

另外一种巧妙的做法：

利用位运算

‘C’ 是 0x43 ‘O’ 是 0x4f ‘W’ 是 0x57

#include <iostream>
#include <stack>
#include <math.h>
#include <cstring>
#include <algorithm>
using namespace std;
int n;
long long c,o,w;
string s;
int main()
{
    cin >> n >> s;
    for(int i=0;i<s.length();i++)
    {
        c += !(s[i] ^ 0x43);
        o += c * !(s[i] ^ 0x4f);
        w += o * !(s[i] ^ 0x57);
    }
    cout<<w<<endl;
    return 0;
}