codeforces 165C Another Problem on Strings 双指针

最新推荐文章于 2020-06-10 21:39:53 发布

原创最新推荐文章于 2020-06-10 21:39:53 发布 · 369 阅读

0 ·

CC 4.0 BY-SA版权

Codeforces 专栏收录该内容

32 篇文章

订阅专栏

本文介绍了一种使用双指针技术解决特定问题的方法：计算一个给定的01字符串中包含恰好k个1的所有子串的数量。通过将每个1的位置及后续0的数量进行记录，并利用乘法原理进行计算，该算法能够高效地解决问题。

题意：求对于给定01字符串，有多少个子串，其恰好包含了k个1

双指针，把每个1都找出来，统计这个1之后有多少个0

然后用乘法原理计算

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define rep(i, j, k) for(int i = j; i <= k; i++)
 
using namespace std;
 
typedef long long lo;

char buff[1001000];
lo l[1001000];

int main() {
	lo sum=0,k, count=0, ptr=0;
	cin >> k >> buff;
	lo n=strlen(buff);
	for (lo i=0;i<n;i++) {
		l[ptr]++;
		if (buff[i]=='1') ptr++;
	}

	l[ptr]++;
	if (k==0) 
		for (lo i=0;i<=ptr;i++)
			(count+=(l[i]-1)*(l[i]-2)/2 + l[i]-1);		
	else 
		for (lo i=0;i+k<=ptr;i++)
			count+=(l[i])*(l[i+k]);
	cout << count;
	return 0;
}