B. Infinite Prefixes

194 篇文章 1 订阅
9 篇文章 0 订阅

链接:https://codeforces.com/contest/1295/problem/B

You are given string ss of length nn consisting of 0-s and 1-s. You build an infinite string tt as a concatenation of an infinite number of strings ss, or t=ssss…t=ssss… For example, if s=s= 10010, then t=t= 100101001010010...

Calculate the number of prefixes of tt with balance equal to xx. The balance of some string qq is equal to cnt0,q−cnt1,qcnt0,q−cnt1,q, where cnt0,qcnt0,q is the number of occurrences of 0 in qq, and cnt1,qcnt1,q is the number of occurrences of 1 in qq. The number of such prefixes can be infinite; if it is so, you must say that.

A prefix is a string consisting of several first letters of a given string, without any reorders. An empty prefix is also a valid prefix. For example, the string "abcd" has 5 prefixes: empty string, "a", "ab", "abc" and "abcd".

Input

The first line contains the single integer TT (1≤T≤1001≤T≤100) — the number of test cases.

Next 2T2T lines contain descriptions of test cases — two lines per test case. The first line contains two integers nn and xx (1≤n≤1051≤n≤105, −109≤x≤109−109≤x≤109) — the length of string ss and the desired balance, respectively.

The second line contains the binary string ss (|s|=n|s|=n, si∈{0,1}si∈{0,1}).

It's guaranteed that the total sum of nn doesn't exceed 105105.

Output

Print TT integers — one per test case. For each test case print the number of prefixes or −1−1 if there is an infinite number of such prefixes.

Example

input

Copy

4
6 10
010010
5 3
10101
1 0
0
2 0
01

output

Copy

3
0
1
-1

Note

In the first test case, there are 3 good prefixes of tt: with length 2828, 3030 and 3232.

题解:赛时太菜,没想过负数(这好像是第二次掉入负数的坑了),赛后面向样例编程QAQ

代码:

#include <bits/stdc++.h>
using namespace std;
long long l,r,n,t,s0,s1,s,ans,k,max1=0;
long long a,b;
char x[10000001];
long long p[10000001];
int main()
{
	cin>>t;
	while(t--)
	{
		cin>>a>>b;
		cin>>x;
		if(b<0)
		{
			b=-b;
			for(int i=0;i<a;i++)
			{
				if(x[i]=='0')
				x[i]='1';
				else
				x[i]='0';
			}
		}
		s=0;
		ans=0;
		int flag=0;
		for(int i=0;i<a;i++)
		{
			if(x[i]=='0')
			s++;
			else
			s--;
			p[i]=s;
			if(s==b)
			{
				flag=1;
			}
		}
		if(b==0)
		ans++;
		if(flag==0&&s==0)
		{
			cout<<0<<endl;
			continue;
		}
		if(flag==1&&s==0)
		{
			cout<<-1<<endl;
			continue;
		}
		if(s!=0)
		{
			for(int i=0;i<a;i++)
			{
				if(b==p[i])
				ans++;
				else
				{
					if(b>p[i]&&s>0)
					{
						if((b-p[i])%s==0)
						ans++;
					}
					else if(b<p[i]&&s<0)
					{
						if((b-p[i])%s==0)
						ans++;
					}
				}
			}
		}
		cout<<ans<<endl;
	}
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值