现在给你一个长度为 n的只有大写字母组成的字符串。字符串里有众多的”WTW“。请你求出字符串里有几个”WTW”,即字符串的子序列为”WTW”的个数。
在数学中,某个序列的子序列是从最初序列通过去除某些元素但不破坏余下元素的相对位置(在前或在后)而形成的新序列。例如A,AB,ACD是字符串ABCD的子序列,而E,BA不是。
Input
第一行给出字符串的长度n(1≤n≤100000)。
第二行是一个长度为n的字符串,字符串保证只有大写字母。
Output
输出一个整数,字符串子序列为”WTW“的个数。所有测试样例的答案保证小于等于1e6。
Sample Input
4 WWTW
Sample Output
2
思路:将字符串存进vector数组中,用find函数去找T的位置,然后再找确定的T的左边有几个W,之后就知道右边有几个W(总W减左边的数量)。(若题目要求找的是WTA这种,就在T右边用多一个循环去找)
#include <bits/stdc++.h>
using namespace std;
int main() {
vector<char>::iterator a;
vector<char>::iterator b;
int n, sum = 0, S1 = 0; char ch;
cin >> n;
vector<char>s;
for (int i = 0; i < n; i++)
{
cin>>ch;
if (ch == 'W')sum++; //算出总共有多少W
s.push_back(ch);
}
a = s.begin();
while( (a = find(a, s.end(), 'T') )!=s.end() )
{
int sum1 = 0;
b = s.begin();
while ( ( b = find(b, a, 'W') ) != a )
//找b到a内的W,就是T之前的
{
sum1++;
b++; //b位置往后,防止重复
}
S1 =S1+ (sum1 * (sum - sum1));
a++;//a位置往后,防止重复
}
cout << S1;
return 0;
}