C. The Number Of Good Substrings
theme:给定一个01串,求子串sl~sr满足sl~sr组成的字符串转化为10进制后的值=r-l+1的个数。1<=|s|<=2e5
solution:首先明确r-l+1最多为2e5,则说明如果字串的第一位为1,那么它最多20为左右就会超2e5,所以可以直接枚举1所在的位置作为第1为,然后只需向后最多枚举20为即可。由于可以有前缀0,所以记录下每一个1前有多少个连续的0.
#include<bits/stdc++.h>
typedef long long ll;
#define far(i,s,n) for(int i=s;i<n;++i)
using namespace std;
struct _a
{
int pos;
int cnt;
_a(int a,int b)
{
pos=a;
cnt=b;
}
};
char a[200020];
vector<_a>v;
int main()
{
int t;
cin>>t;
while(t--)
{
scanf("%s",a);
v.clear();
int n=strlen(a);
int cnt=0;
far(i,0,n)
{
if(a[i]=='1')
{
v.push_back(_a(i,cnt));
cnt=0;
}
else
++cnt;
}
int sz=v.size();
int ans=0;
far(i,0,sz)
{
int start=v[i].pos;
int cnt=v[i].cnt;
int num=0;
for(int j=start;j<start+20&&j<n;++j)
{
num=num*2+a[j]-'0';
if(num==j-start+1)
++ans;
else if(num>j-start+1&&cnt>=num-j+start-1)
++ans;
}
}
printf("%d\n",ans);
}
}