Skip to content

fix: correct treap split() comparison to match docstring behavior - #14762

Merged
cclauss merged 4 commits into
TheAlgorithms:masterfrom
maitriupadhyay03-cell:master
Sep 11, 2026
Merged

fix: correct treap split() comparison to match docstring behavior#14762
cclauss merged 4 commits into
TheAlgorithms:masterfrom
maitriupadhyay03-cell:master

Conversation

@maitriupadhyay03-cell

Copy link
Copy Markdown
Contributor

Describe your change:

The split() function in data_structures/binary_tree/treap.py uses a strict < comparison, but the docstring states:

Left tree contains all values less than split value.
Right tree contains all values greater or equal, than split value

This means when splitting by a value that already exists in the treap, the existing node ends up in the LEFT subtree instead of the RIGHT subtree, which contradicts the documented behavior.

Fix: Changed elif value < root.value: to elif value <= root.value: so that equal values go to the right subtree as documented.

Fixes #7854

  • Add an algorithm?
  • Fix a bug or typo in an existing algorithm?
  • Add or change doctests? -- Note: Please avoid changing both code and tests in a single pull request.
  • Documentation change?

Checklist:

  • I have read CONTRIBUTING.md.
  • This pull request is all my own work -- I have not plagiarized.
  • I know that pull requests will not be merged if they fail the automated tests.
  • This PR only changes one algorithm file. To ease review, please open separate PRs for separate algorithms.
  • All new Python files are placed inside an existing directory.
  • All filenames are in all lowercase characters with no spaces or dashes.
  • All functions and variable names follow Python naming conventions.
  • All function parameters and return values are annotated with Python type hints.
  • All functions have doctests that pass the automated testing.
  • All new algorithms include at least one URL that points to Wikipedia or another similar explanation.
  • If this pull request resolves one or more open issues then the description above includes the issue number(s) with a closing keyword: "Fixes #ISSUE-NUMBER".

…gorithms#7854)

The split() function in treap.py uses `<` comparison but the docstring states that the right subtree should contain values "greater or equal" to the split value. This fix changes `elif value < root.value:` to `elif value <= root.value:` so that equal values go to the right subtree as documented.

Fixes TheAlgorithms#7854
@algorithms-keeper

Copy link
Copy Markdown

Closing this pull request as invalid

@maitriupadhyay03-cell, this pull request is being closed as none of the checkboxes have been marked. It is important that you go through the checklist and mark the ones relevant to this pull request. Please read the Contributing guidelines.

If you're facing any problem on how to mark a checkbox, please read the following instructions:

  • Read a point one at a time and think if it is relevant to the pull request or not.
  • If it is, then mark it by putting a x between the square bracket like so: [x]

NOTE: Only [x] is supported so if you have put any other letter or symbol between the brackets, that will be marked as invalid. If that is the case then please open a new pull request with the appropriate changes.

@algorithms-keeper algorithms-keeper Bot closed this Jun 6, 2026
@algorithms-keeper algorithms-keeper Bot added the awaiting reviews This PR is ready to be reviewed label Jun 6, 2026
@cclauss cclauss reopened this Sep 10, 2026
@algorithms-keeper algorithms-keeper Bot added enhancement This PR modified some existing files tests are failing Do not merge until tests pass labels Sep 10, 2026
@priya-sundaram-dev

Copy link
Copy Markdown
Contributor

Thanks for this @maitriupadhyay03-cell — the split change is correct and finally matches the docstring. 🙌

One thing to finish before it can close #7854: this one-liner breaks erase, which was written against the old split semantics (a known gotcha noted on the issue back in 2022). The current interact_treap doctest fails after your change:

-0  → Got 0,1,2,...   (0 not erased)
-4  → Got 0,1,2,4,4,4,4,5,...

Updating erase to the new semantics fixes it and makes all doctests pass:

def erase(root: Node | None, value: int) -> Node | None:
    left, right = split(root, value)
    _, right = split(right, value + 1)
    return merge(left, right)

With both split and erase updated, python -m doctest treap.py -v is green. 👍

Adjust split logic to include the value in the right subtree.
@algorithms-keeper algorithms-keeper Bot removed the tests are failing Do not merge until tests pass label Sep 11, 2026
@algorithms-keeper algorithms-keeper Bot removed the awaiting reviews This PR is ready to be reviewed label Sep 11, 2026
@cclauss
cclauss merged commit 3d11d0c into TheAlgorithms:master Sep 11, 2026
6 checks passed
@priya-sundaram-dev

Copy link
Copy Markdown
Contributor

Nice — that's exactly it. ✅ Pulled your branch and ran python -m doctest treap.py -v: 11 tests in 12 items, 11 passed / 0 failed. Both split and erase now line up with the docstring, and the interact_treap doctest (the -0 / -4 erase cases) is green again. This should be good to close #7854. Thanks for following through! 🙌

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement This PR modified some existing files invalid

Projects

None yet

Development

Successfully merging this pull request may close these issues.

treap not working expected when splitting on value that existed

3 participants