2019年湘潭大学程序设计竞赛(重现赛) F 二分+前缀和

https://ac.nowcoder.com/acm/contest/893/F
首先预处理出前缀和,a[i]表示这个区间有多少个1

然后二分答案,最对答案进行O(n)验证

只需判断区间内0或1的个数加上m是否不小于当前二分的答案即可 时间复杂度O(nlongn)

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<map> 
#include<set>
#include<deque>
#include<queue>
#include<stack>
#include<bitset>
#include<string>
#include<fstream>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
string s;
int a[100001];
int n,m;
int find(int x)
{
	for(int i=0;i<=s.size()-x;i++)
	{
		int num=a[i+x-1]-a[i]+s[i]-'0';
		int maxx=max(num,x-num);// 找这段区间内 1 和 0个数的最大值 
		if(maxx+m>=x)
		return 1;
	}
	return 0;
}
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n>>m;
		cin>>s;
		a[0]=s[0]-'0';
		for(int i=1;i<s.size();i++)
		{
			a[i]=a[i-1]+s[i]-'0';// 前缀中1的个数 
		}
		int l=1,r=s.size();
		int ans=0;
		while(l<=r)
		{
			int mid=(l+r)/2;
			if(find(mid))
			{
				ans=mid;
				l=mid+1;
			}
			else
			r=mid-1;
		} 
		cout<<ans<<endl;
		
	}
	return 0;
}

 

展开阅读全文

没有更多推荐了,返回首页