【51nod-1396】还是01串

给定一个0-1串s，长度为n，下标从0开始，求一个位置k，满足0<=k<=n, 并且子串s[0..k - 1]中的0的个数与子串s[k..n - 1]中1的个数相等。 注意：

（1） 如果k = 0, s[0..k - 1]视为空串

（2） 如果k = n, s[k..n - 1]视为空串

（3） 如果存在多个k值，输处任何一个都可以

（4） 如果不存在这样的k值，请输出-1

Input

就一行，包含一个0-1串S，长度不超过1000000。

Output

题目要求的k值

Input示例

01

Output示例

1

可以先从结论入手，假设在k处断开，前面有x个1，那么就有k - x个0，后面设有y个1，由题意得k-x = y, k = x+y，即求出1的个数即可

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1000005;
char s[N];
int main()
{
    scanf("%s", s);
    int l = strlen(s), num = 0;
    for(int i = 0; i < l; i++)
    {
        if(s[i] == '1')
            num++;
    }
    printf("%d
", num);
    return 0 ;
}