WOW Factor CodeForces - 1178B (思维,预处理,排列组合)

题意:给一个长度为N仅由小写字母'o'和'v'组成的字符串S 求S中子序列‘vv’-‘o’-‘vv’的个数。 注意:vv必须连续,而vv和o,o和vv之间不一定连续

题解:对于单独的一个‘o’来说wow的个数等于左右两边w个数的乘积,用两个数组预处理出来每个位置左右两边的ww个数即可。

AC代码:

#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <algorithm>
#define int long long
using namespace std;
const int maxn=1e6+5;
int l[maxn],r[maxn];
main(){
	string s;
	cin>>s;
	for(int i=1;i<s.size();i++){
		if(s[i]=='v'&&s[i-1]=='v')l[i]=l[i-1]+1;
		else l[i]=l[i-1];
	}
	for(int i=s.size()-1;i>=1;i--){
		if(s[i]=='v'&&s[i-1]=='v')r[i]=r[i+1]+1;
		else r[i]=r[i+1];
	}
	int ans=0;
	for(int i=0;i<s.size();i++){
		if(s[i]=='o')ans+=r[i]*l[i];
		//cout<<i<<" "<<l[i]<<" "<<r[i]<<endl; 
	}
	cout<<ans<<endl;
}

猜你喜欢

转载自blog.csdn.net/Alanrookie/article/details/107516008