1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
|
#include <iostream>
#include <vector>
using namespace std;
struct node
{
int data;
node *left = 0, *right = 0;
};
void insert( node *n, bool add0, bool add1 )
{
if ( n->left || n->right ) // NOT at bottom of tree, don't add branches
{
if ( n->left ) insert( n->left , add0, add1 );
if ( n->right ) insert( n->right, add0, add1 );
}
else // at bottom of tree, so CAN add branches
{
if ( add0 ) n->left = new node{ 0, 0, 0 };
if ( add1 ) n->right = new node{ 1, 0, 0 };
}
}
void output( node *n )
{
if ( !n ) return;
output( n->left );
cout << n->data;
output( n->right );
}
int main()
{
vector<int> A = { 6, 4, 1, -2 };
node *root = new node{ 0, 0, 0 };
for ( int e : A ) insert( root, e >= 0, e < 0 || e >= 3 ); // add 0 if e is positive or zero
// add 1 if e is negative or >= 3
output( root );
}
| |