The 2019 ICPC China Shaanxi Provincial Programming Contest E. Turn It Off

训练赛写的这道题，看了半天没有思路，卡了很久。后来实在没啥思路，就直接暴力了，还以为会T，结果过了，不过赛后看题解发现是个二分。

整理一下思路吧。

题目链接：Problem - E - Codeforces

题目大意：有一排n数量的灯，有开有关的，每次可以关连续的L里的灯，最多可以关k次，问每次关的距离最短可以是多少。

题目思路：

1.暴力：我开始的做法，记录一下字符串里面1的数量num，如果num比可以关的次数小或者等于，那么就直接输出1，否则根据让num/k向上取整，这样就得到了一个最小的区间长度mi，使用这个区间长度遍历，看是否能遍历完，不能就让mi++，直到可以为止。

代码如下：

#include<iostream>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<stdio.h>
#include<vector>
using namespace std;
int t,n,k;
string s;
int main()
{
	scanf("%d",&t);
	while(t--)
	{
		int num=0,cnt=0;
		scanf("%d%d",&n,&k);
		cin>>s;
		for(int i=0;i<n;i++)
		{
			if(s[i]=='1')num++;
		}
		for(int i=s.size()-1;i>=0;i--)
		{
			if(s[i]=='1')break;
			n--;
		}
		if(num<=k)printf("1\n");
		else
		{
			bool pd=0;
			int mi=ceil((float)num/k);
			while(1)
			{
				int i;
				cnt=0;
				for(i=0;i<n;)
				{
					if(s[i]=='0')
					{
						i++;
						continue;
					}
					if(s[i]=='1')
					{
						cnt++;
						i+=mi;
					}
					if(cnt==k)break;
				}		
				if(i>=n)break;
				mi++;	
			}
			printf("%d\n",mi);
		}
	}
	return 0;
}

2.二分法：前面跟上面思路一样，记录一下字符串里面1的数量num，如果num比可以关的次数小或者等于，那么就直接输出1，否则二分出合法的最小区间长度。

代码如下：

#include<iostream>
#include<algorithm>
#include<string.h>
#include<stdio.h>
#include<math.h>
using namespace std;
int t,n,k;
string s;

bool check(int x)
{
	int pd=0,cnt=0,num=0;
	for(int i=1;i<=n;i++)
	{
		if(!pd&&s[i]=='1')
		{
			num++;
			pd++;
			cnt++;
		}
		else if(pd&&cnt<x)
		{
			cnt++;
		}
		if(pd&&cnt==x)
		{
			pd=0;
			cnt=0;
		}
	}
	if(num<=k)return 1;
	else return 0;
}

int main()
{
	cin>>t;
	while(t--)
	{
		int num=0;
		cin>>n>>k;
		cin>>s;
		s=" "+s;
		for(int i=1;i<=n;i++)
		{
			if(s[i]=='1')num++;
		}
		if(num<=k)cout<<"1"<<endl;
		else
		{
			int l=1,r=2e5+10;
			while(l<r)
			{
				int mid=(l+r)/2;
				if(check(mid))
				{
					r=mid;
				}
				else
				{
					l=mid+1;
				}
			}
			cout<<l<<endl;
		}
	}
	return 0;
}

--------------------------------------------------------END---------------------------------------------------------