贡献法:
贡献法就是计算每个元素对最终答案的贡献是多少,在枚举的过程中加起来。所以这类题目的关键是想到如何在枚举的过程中计算各个元素的贡献。
题目描述:
Farmer John 最近购入了 N 头新的奶牛,每头奶牛的品种是更赛牛(Guernsey)或荷斯坦牛(Holstein)之一。
奶牛目前排成一排,Farmer John 想要为每个连续不少于三头奶牛的序列拍摄一张照片。
然而,他不想拍摄这样的照片,其中只有一头牛的品种是更赛牛,或者只有一头牛的品种是荷斯坦牛——他认为这头奇特的牛会感到孤立和不自然。
在为每个连续不少于三头奶牛的序列拍摄了一张照片后,他把所有「孤独的」照片,即其中只有一头更赛牛或荷斯坦奶牛的照片,都扔掉了。
给定奶牛的排列方式,请帮助 Farmer John 求出他会扔掉多少张孤独的照片。
如果两张照片以不同位置的奶牛开始或结束,则认为它们是不同的。
输入格式
输入的第一行包含 N。
输入的第二行包含一个长为 N 的字符串。如果队伍中的第 i 头奶牛是更赛牛,则字符串的第 i 个字符为 G
。否则,第 i 头奶牛是荷斯坦牛,该字符为 H
。
输出格式
输出 Farmer John 会扔掉的孤独的照片数量。
数据范围
3≤N≤5×10^5
输入样例:
5
GHGHG
输出样例:
3
样例解释
这个例子中的每一个长为 33 的子串均恰好包含一头更赛牛或荷斯坦牛——所以这些子串表示孤独的照片,并会被 Farmer John 扔掉。
所有更长的子串(GHGH
、HGHG
和 GHGHG
)都可以被接受。
解题思路:
暴力法:直接采用两次for循环O(n*n) ,在循环里面进行计算,普通的查找需要复杂度为O(n),采用前缀和的话为O(1),对于这道题目来说差不多得一半的分。
贡献法:我们从每一头牛对这个孤独的照片总数的贡献进行计算,这样求一头牛对这个孤独的照片总数的贡献的复杂度为O(1),最后的复杂度为O(n)。对于每一头牛来说它对这个孤独的照片,有三种情况。
第一种:左右两边都有和自己不一样的牛数量为L和R,贡献为L*R+R-1+L-1。
第二种:左边没有,右边有并且大于2,贡献为R-1。
第三种:右边没有,左边有并且大于2,贡献为L-1。
代码及代码解释:
#include<stdio.h>
#define N 500010
typedef long long LL;//因为某些结果会超出int的范围,所以结果用long long 类型数据存储
int n,l[N],r[N];//两个数组分别用于存储左右两边和它不一样的连续字符的个数
char arr[N];
int max(int m,int n)//进行比较,保存不会出现负数
{
if(m<n)return n;
else return m;
}
int main()
{
int sh,sg;
scanf("%d",&n);
scanf("%s",arr);
for(int i=0,sh=0,sg=0;i<n;i++)//计算这个数据的左边连续的和他不一样的字符个数,并保存在数组里面
if(arr[i]=='G')l[i]=sh,sg++,sh=0;
else l[i]=sg,sh++,sg=0;
for(int i=n-1,sh=0,sg=0;i>=0;i--)//计算这个数据的右边连续的和他不一样的字符个数,并保存在数组里面
if(arr[i]=='G')r[i]=sh,sg++,sh=0;
else r[i]=sg,sh++,sg=0;
LL ans=0;
for(int i=0;i<n;i++)//统计结果
ans+=(LL)l[i]*r[i]+max(0,r[i]-1)+max(0,l[i]-1);//不论是哪一种情况,最后结果都可以用这个表达式进行计算
printf("%lld",ans);
return 0;
}