字符串 AAZBCZCUCIOCQWC 共包含6个 ZUCC。
例如,第3位(Z),第8位(U),第9位©,第12位©可以形成一个ZUCC;
第6位(Z),第8位(U),第12位©,第15位©也可以形成一个 ZUCC。
下面给定一个长度不超过 105 且仅包含大写 A-Z 字母的字符串。一共可以形成多少个 ZUCC?
输入格式:
输入只有一行,包含一个长度不超过 105 的字符串。
输出格式:
输出该字符串中包含的 ZUCC 个数。题目保证总数不超过 263−1。
输入样例:
AAZBCZCUCIOCQWC
输出样例:
6
从最后一个字符向前看,用c来记录表示找到字符C的个数,每找到一个字符C,那么c+1。当找到一个U时,后面的C的排列方式可以用简单的数学排列计算得出,如c=3,则为3个C中任意取出两个C,c*(c-1)/2=3种组合方式(若C的个数为1个或0个则算出来的组合方式为0种)。此时在找到的这个U处设置一个值m(m 的初始值为0)记录此个U后c的组合方式即m +=c*(c-1)/2。再从此U向前继续遍历字符,在遍历过程中找到C或者U时的处理和前面叙述相同,若找到便更新c , m 的值,直到找到字符Z,用sum来表示总的ZUCC个数,sum 的初始值为0,sum += m。此后继续向前遍历,指导遍历完成,遇到ZUCC中的字符都按上述操作进行啦。
#include<stdio.h>
#include<string.h>
int main()
{
int c=0,n,i;
char s[100001];
long long m=0;
long long sum=0;
scanf("%s",s);
n=strlen(s);
for(i=n-1;i>=0;i--){
if(s[i]=='C')
c++;
else if(s[i]=='U')
m+=c*(c-1)/2;
else if(s[i]=='Z')
sum+=m;
}
printf("%lld\n",sum);
return 0;
}