public bool FindTarget(TreeNode root, int k)
    {
        BSTIterator nextIterator = new BSTIterator(root),
                    preIterator  = new BSTIterator(root, true);

        for (TreeNode left = nextIterator.HasNext() ? nextIterator.Next() : null,
             right = preIterator.HasNext()? preIterator.Next() : null; left.val <
             right.val;)
        {
            int sum = left.val + right.val;
            if (sum == k)
            {
                return(true);
            }
            if (sum < k)
            {
                left = nextIterator.HasNext()? nextIterator.Next() : null;
            }
            else
            {
                right = preIterator.HasNext()? preIterator.Next() : null;
            }
        }
        return(false);
    }
Ejemplo n.º 2
0
        public void BSTIteratorTest()
        {
            TreeNode root = new TreeNode(2);

            root.left  = new TreeNode(1);
            root.right = new TreeNode(3);

            BSTIterator bst = new BSTIterator(root);
        }
Ejemplo n.º 3
0
        public void Run()
        {
            var iterator = new BSTIterator(Utility.ConvertToTreeNode("[7,3,15,null,null,9,20]"));

            iterator.Next();
            iterator.Next();
            iterator.Next();
            iterator.Next();
            iterator.Next();
            iterator.HasNext();
        }
Ejemplo n.º 4
0
        private void Check(TreeNode root)
        {
            var it = new BSTIterator(root);

            var v = 0;

            while (it.HasNext())
            {
                var n = it.Next();
                Assert.True(v < n);
                v = n;
            }
        }
Ejemplo n.º 5
0
        public void NextTest()
        {
            TreeNode root = new TreeNode(2);

            root.left  = new TreeNode(1);
            root.right = new TreeNode(3);

            BSTIterator bst = new BSTIterator(root);

            Assert.IsTrue(bst.Next() == 1);
            Assert.IsTrue(bst.Next() == 2);
            Assert.IsTrue(bst.Next() == 3);
        }
Ejemplo n.º 6
0
        public void Test1()
        {
            BSTIterator iterator = new BSTIterator(_root);
            var         n1       = iterator.Next();    // return 3
            var         n2       = iterator.Next();    // return 7
            var         hasNext1 = iterator.HasNext(); // return true
            var         n3       = iterator.Next();    // return 9
            var         hasNext2 = iterator.HasNext(); // return true
            var         n4       = iterator.Next();    // return 15
            var         hasNext3 = iterator.HasNext(); // return true
            var         n5       = iterator.Next();    // return 20
            var         hasNext4 = iterator.HasNext(); // return false

            Assert.Pass();
        }
Ejemplo n.º 7
0
    public static bool Run(TreeNode p, TreeNode q)
    {
        if (p == null || q == null)
        {
            return(false);
        }

        var pIt = new BSTIterator(p);
        var qIt = new BSTIterator(q);

        while (pIt.HasNext() && qIt.HasNext())
        {
            if (pIt.Next() != qIt.Next())
            {
                return(false);
            }
        }

        return(!pIt.HasNext() && !qIt.HasNext());
    }
Ejemplo n.º 8
0
        public static void Run()
        {
            var st = new Solution173();

            Console.WriteLine("Start: {0}", DateTime.Now);

            TreeNode root = new TreeNode(10);
            root.left = new TreeNode(5);
            root.right = new TreeNode(20);
            root.left.left = new TreeNode(1);
            root.left.right = new TreeNode(8);
            root.left.right.left = new TreeNode(7);
            root.right.right = new TreeNode(21);

            BSTIterator i = new BSTIterator(root);
            while (i.HasNext())
                Console.WriteLine(i.Next());

            Console.WriteLine("End: {0}", DateTime.Now);
        }
Ejemplo n.º 9
0
        public void Case1()
        {
            var input = new TreeNode(7);

            input.left        = new TreeNode(3);
            input.right       = new TreeNode(15);
            input.right.left  = new TreeNode(9);
            input.right.right = new TreeNode(20);

            var target = new BSTIterator(input);

            Assert.Equal(3, target.Next());
            Assert.Equal(7, target.Next());
            Assert.True(target.HasNext());
            Assert.Equal(9, target.Next());
            Assert.True(target.HasNext());
            Assert.Equal(15, target.Next());
            Assert.True(target.HasNext());
            Assert.Equal(20, target.Next());
            Assert.False(target.HasNext());
        }
Ejemplo n.º 10
0
        public void BSTIterator_Test()
        {
            var iterator = new BSTIterator(BsTree);

            if (iterator.hasNext())
            {
                var a = iterator.next();

                Assert.Equal(2, a);

                var b = iterator.next();

                Assert.Equal(4, b);

                var c = iterator.next();

                Assert.Equal(5, c);

                var hasNext = iterator.hasNext();

                Assert.True(hasNext);
            }
        }
Ejemplo n.º 11
0
        public void DataStructure_BSTIteratorTest1()
        {
            TreeNode root = new TreeNode(4),
           l  = new TreeNode(2),
           r  = new TreeNode(5),
           ll = new TreeNode(1),
           lr = new TreeNode(3);

            root.Left  = l;
            root.Right = r;
            l.Left     = ll;
            l.Right    = lr;

            BSTIterator it = new BSTIterator(root);

            Assert.IsTrue(it.HasNext());
            Assert.AreEqual(1, it.Next());
            Assert.AreEqual(2, it.Next());
            Assert.AreEqual(3, it.Next());
            Assert.AreEqual(4, it.Next());
            Assert.AreEqual(5, it.Next());
            Assert.IsFalse(it.HasNext());
        }