51nod oj 1396 还是01串【打表】

传送门：1396

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
char ch[1010000];
int zero[1010000],one[1010000];
int main()
{
	scanf("%s",ch+1);
	int ll=strlen(ch+1);
	zero[0]=0;
	for (int i=1;i<=ll;i++)
	{
		zero[i]=zero[i-1];
		if (ch[i]=='0') zero[i]++;
	}
	one[ll+1]=0;
	for (int i=ll;i>=1;i--)
	{
		one[i]=one[i+1];
		if (ch[i]=='1') one[i]++;
	}
	bool fafe=true;
	for (int i=0;i<=ll;i++)
	if (zero[i]==one[i+1])
	{
		fafe=false;
		printf("%d\n",i);
		break;
	}
	if (fafe) printf("-1\n");
	return 0;
}