(第十四届蓝桥真题)字串简写

 样例输入:

4
abababdb a b

样例输出:

6

分析:

 这道题还是比较简单的,我们需要记录字符c1和c2出现的位置,对于每一个c2出现的位置,我们都可以寻找在其前面且距离大于等于k的字符1的出现次数,这个可以直接二分求解,也可以用前缀和进行优化,复杂度都是可以的。需要注意的一点是c1和c2有可能相同。

细节见代码:

#include
#include
#include
#include
#include
#include
using namespace std;
const int N=5e5+10;
char s[N];
int p[N],tt;
int main()
{
	int k;
	scanf("%d",&k);
	scanf("%s",s+1);
	int n=strlen(s+1);
	char c1[2],c2[2];
	scanf("%s%s",c1,c2);
	long long ans=0;
	for(int i=1;i<=n;i++)
	{
		if(c1[0]==s[i]) p[++tt]=i;
		if(c2[0]==s[i])
		{
			if(!tt||i-p[1]+1>1;
				if(i-p[mid]+1>=k) l=mid;
				else r=mid-1;
			}
			ans+=l;
		}
	}
	printf("%lld",ans);
	return 0;
}

你可能感兴趣的:(蓝桥杯,职场和发展)