这道题不难,但是还是想了好久。。。(太菜了)
思路:
设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;
}