51nod 1396还是01串(前缀和)

1396 还是01串

基准时间限制：1 秒 空间限制：131072 KB 分值: 20 难度：3级算法题 收藏 关注

给定一个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

题解：

做过类似的题自然会有经验的，01串转化为10进制，进行前缀和计算都是常用的技巧。这里只需要计算前缀和pre[ k]代表0-k-1的数字和，那么如果符合以下的条件就是符合题意的：k-pre[k] == sum-pre[k]，sum代表所有的数字之和。

注意点：不能使用cin，会超时。一般输入量大于1e5的时候就不能使用cin了

代码：

#include<iostream>
#include<fstream>
#include<string.h>
#include<math.h>
#include<stdlib.h>
#include<stdio.h>
#include<utility>
#include<algorithm>
#include<map>
#include<stack>
#include<set>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn = 1e6+5;
const int mod = 1e9+7;
const int INF = 1<<30;
const ll llINF = 1e18+999;

int arr[maxn], pre[maxn], sum, i, n;
char in[maxn];
void init( )
{
    n = strlen(in);
    pre[0] = 0;
    for(int i=0; i<n; i++)
        arr[i] = in[i]-'0';
    for(int i=1; i<=n; i++)
        pre[i] = pre[i-1]+arr[i-1];
    sum = pre[n];
}
int main( )
{
    //freopen("input.txt", "r", stdin);
    //freopen("output.txt", "w", stdout);
    while(~scanf("%s", in))
    {
        init( );
        for(i=0; i<=n; i++)
            if(i-pre[i] == sum-pre[i])
                break;
        if(i <= n)
            cout<<i<<endl;
        else
            cout<<-1<<endl;
    }
    return 0;
}