Optimized Code for 1's and 0's

opt_1s_0s.cpp

int diff = 0;

for (int i = 0; i <= n; i++){
   if (B[diff + n] == -1) B[diff + n] = i; 
   else C[diff + n] = i;
   if (A[i]) diff++; else diff--;     
}