吉首大学2019年程序设计竞赛（重现赛）,G（思维）

这道题不难，但是还是想了好久。。。（太菜了）
思路：
设po[i]，记录第i个O的位置；
设sum0[i]，记录s[0~i]中有多少个Z；
设sum[i],记录s[i~len-1]有多少组RZ；
最终答案：SUM(sum[po[1~cnt])

#include<iostream>
#include<stack>
#include<cstdio>
#include<queue> 
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e5+5;
char s[maxn];
int po[maxn],sum[maxn],sum0[maxn];

int main()
{
	scanf("%s",s);
	int len=strlen(s);
	memset(po,0,sizeof(po));
	memset(sum0,0,sizeof(sum0));
	memset(sum,0,sizeof(sum));
	int cnt=1,o=1;
	for(int i=0;i<len;++i)
	{
		if(i)	sum0[i]+=sum0[i-1];
		if(s[i]=='O')
			po[o++]=i;
		else if(s[i]=='Z')	sum0[i]++; 
	}
	long long ans=0;
	for(int i=len-1;i>=0;--i)
	{
		sum[i]=sum[i+1];
		if(s[i]=='R')
			sum[i]+=(sum0[len-1]-sum0[i]);
	}
	for(int i=1;i<o;++i)	ans+=sum[po[i]];
	printf("%lld\n",ans);
	return 0;
}