2272: 又是01串
题目描述
给定一个0-1串s,长度为n,下标从0开始,求一个位置k,满足0<=k<=n, 并且子串s0..k−1中的0的个数与子串sk..n−1中1的个数相等。 注意:
(1) 如果k = 0, s0..k−1视为空串
(2) 如果k = n, sk..n−1视为空串
(3) 如果存在多个k值,输出任何一个都可以
(4) 如果不存在这样的k值,请输出-1
输入
就一行,包含一个0-1串S,长度不超过1000000。
输出
题目要求的k值
样例输入
01
样例输出
1
解题思路:
计算出字符串中字符为1的个数d, 假设从k处断开, 此时后面有x个1,那么前面就有d-x个1,同时前面有x个0(因为0的个数与1的个数相同), k = d-x + x = d, 所以k的位置,就是1的个数(特别的k可以为0和n)。
代码如下:
#include<stdio.h>
#include<string.h>
char s[1001000];
int main()
{
int i,d=0,len;
scanf("%s",s);
len=strlen(s);
for(i=0;i<len;i++)
{
if(s[i]=='1')
d++;
}
if(d>len)//k值不存在的情况
printf("-1\n");
else
printf("%d\n",d);
return 0;
}