sort_array_containing_zero_and_one.cpp

// Write a C++ program to sort an array containing only 0 and 1
// to run program 
// g++ -o sort_array_containing_zero_and_one sort_array_containing_zero_and_one.cpp
// ./sort_array_containing_zero_and_one 

#include<iostream>
using namespace std;


int main()
{

    int A[100], n, i, tmp, nb_zero;

	cout<<"Enter number of elements you want to insert ";
	cin>>n;

    nb_zero = 0;
    i = 0;

    while (i < n)
    {
		cout<<"Enter element "<<i+1<<":";
		cin>>tmp;
        if ((tmp == 0) || (tmp == 1))
        {
            if (tmp == 0)
            {
                nb_zero += 1;
            }
            A[i] = tmp;
            i+=1;
        } else 
        {
            cout<<"Please enter 0 or 1\n";
        }
    }


    // print array
    cout << "\nYour initial array :";
    for(i=0;i<n;i++) 
    {
        cout << A[i] << " ";
    }

    // sorted array
    for(i=0; i<n;i++)
    {
        tmp = i < nb_zero? 0 : 1;
	A[i] = tmp;
    }

    // print sorted array
    cout << "\nYour array sorted: ";
    for(i=0; i<n;i++)
    {
      cout << A[i] << " ";
    }

}