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; }