一个个找区间会T
可以找完第一个区间之后把区间往后平移,调整区间的大小
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll t,n,m,j;
string s;
int main()
{
cin>>t;
while(t--)
{
ll sum=0;
cin>>n>>m>>s;
ll x=0,y=0;
for(int i=0;i<n;i++)
{
if(s[i]=='0')
{
x++;
}
if(x>m) break;
y++;
}
sum+=y;
if(x>m) x=m;
//cout<<y<<endl;
for(int i=1;i<n;i++)
{
if(i+y-1<n)
{
//cout<<1<<endl;
y--;
if(s[i-1]=='0') x--;
for(j=i+y;j<n;j++)
{
if(s[j]=='0') x++;
if(x>m)
{
x--;
break;
}
y++;
}
sum+=y;
//cout<<i<<" "<<y<<endl;
}
else
{
//cout<<1<<endl;
sum=sum+n-i;
}
//cout<<y<<endl;
}
cout<<sum<<endl;
}
return 0;
}