CodeCraft-22 and Codeforces Round 795 (Div. 2)C. Sum of Substrings(分类讨论、贪心)

感觉分类讨论的能有点弱。遇到复杂一点的分类讨论的题目，代码就写的巨长。
首先观察到处在中间位置的1对答案的贡献是11，具体在中间哪个位置是没有关系的。
只有两端的两个位置是比较特殊的
$1\mathrm{位}\mathrm{置}\mathrm{处}\mathrm{的}1\mathrm{对}\mathrm{答}\mathrm{案}\mathrm{的}\mathrm{贡}\mathrm{献}\mathrm{是}10$
$2\mathrm{位}\mathrm{置}\mathrm{处}\mathrm{的}1\mathrm{对}\mathrm{答}\mathrm{案}\mathrm{的}\mathrm{贡}\mathrm{献}\mathrm{是}1$
所有我们考虑将最左端第一次出现的1放到1位置
将最右端第一次出现的1放到n的位置。
贪心的考虑，如果能进行第二个操作我们优先进行第2个操作，因为我们希望答案最小，不能进行第2个操作的情况下我们进行第1种操作。
$\mathrm{操}\mathrm{作}\mathrm{完}\mathrm{后}\mathrm{如}\mathrm{果}1\mathrm{位}\mathrm{置}\mathrm{能}\mathrm{有}1\mathrm{我}\mathrm{们}\mathrm{就}\mathrm{将}\mathrm{答}\mathrm{案}-1$
$\mathrm{如}\mathrm{果}n\mathrm{位}\mathrm{置}\mathrm{能}\mathrm{有}1\mathrm{我}\mathrm{们}\mathrm{就}\mathrm{将}\mathrm{答}\mathrm{案}-10$
代码很丑$qwq$

#include <bits/stdc++.h> 
#define int long long
#define rep(i,a,b) for(int i = (a); i <= (b); ++i)
#define fep(i,a,b) for(int i = (a); i >= (b); --i)
#define ls p<<1
#define rs p<<1|1
#define pii pair<int, int>
#define pll pair<long long, long long>
#define ll long long
#define ull unsigned long long
#define db double
#define endl '\n'
#define debug(a) cout<<#a<<"="<<a<<endl;
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define INF 0x3f3f3f3f 
#define x first
#define y second
#define pb push_back

using namespace std;
const int N=2e5+10;
char a[N];
void solve()
{
	int n,k;cin>>n>>k;
	cin>>(a+1);
	int l1=1,r1=n,cnt=0;
	rep(i,1,n)	if(a[i]=='1')	cnt++;
	//找左边第1个1的位置
	rep(i,1,n) if(a[i]=='1')	
	{
		l1=i;
		break;	
	}		
	
	//找右边第一个1的位置
	fep(i,n,1)	if(a[i]=='1')	
	{
		r1=i;
		break;
	}

	if(cnt==0)
	{
		cout<<0<<endl;
		return;
	}
	//只有1个1
	else if(cnt==1)
	{
		int dl=l1-1,dr=n-r1;
		//先判是否能道右边
		if(k>=dr)
		{
			cout<<1<<endl;
			return;
		}
		//左边
		if(k>=dl)
		{
			cout<<10<<endl;
			return;
		}
		else	out<<11<<endl;
	}
	else
	{
		int dl=l1-1,dr=n-r1;
		int ans=11*cnt;	
		//先判是否能到右边
		if(k>=dr)	ans-=10,k-=dr;
		//左边
		if(k>=dl)	ans-=1;
		cout<<ans<<endl;
	}
	
}
signed main()
{
	IOS	
//  	freopen("1.in", "r", stdin);
  	int _;
	cin>>_;
	while(_--)
	solve();
	return 0;
}