01字符串

B
题意:给出一个01字符串s,将它无限重复,求有几个前缀和为x的前缀(0则+1,1则-1)
进行+1,-1的操作,最后的值肯定是从最小的值一个个过渡到最大的值
1.题中给出:无限个则为-1,前提肯定是能满足x的,那么即整个串为0,串的前缀又符合为x。
2.满足x的:即x=num*()+Min~Max。
3.否则不满足x,即为0。
注意:空前缀只有在x=0时才有贡献

#include<iostream>
#include<cstdio>
#include<string>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<map>
#include<vector>
using namespace std;
#define ll long long
ll sum[1000010];
signed main()
{
	int T; cin>>T;
	while(T--){
		map<ll,ll>mp;
		ll n,x; cin>>n>>x;
		string s; cin>>s;
		ll num=0,Max=0,Min=n;
		for(ll i=0;i<n;i++){
			if(s[i]=='0') num++;
			else num--;
			Max=max(Max,num);
			Min=min(Min,num);
			mp[num]++;
		}
		ll ans=0;
		int flag=0,fla=0;
		for(ll i=Min;i<=Max;i++){
			if(i==x){
				flag=1;
				break;
			}
		}
		if(num==0){
			if(flag==1) ans=-1;
			else{
				ans=0;
				flag=1;
			}
		}
		else{
			if(x==0){
				ans=1;
				fla=1;
			}
//			if(Min<=x&&x<=Max){
//				ans+=mp[x];
//				fla=1;
//			}
//			else{
				for(ll i=Min;i<=Max;i++){
					if((x-i)*num>=0&&(x-i)%num==0){
						ans+=mp[i];
						fla=1;
//						cout<<i<<" ";
					}
				}
//			}
		}
		if(flag==0&&fla==0) ans=0;
		cout<<ans<<endl;
	} 
} 
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值