-
-
Notifications
You must be signed in to change notification settings - Fork 46.6k
Added Fibonacci_Heap for DSA #9285
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from 4 commits
1b296aa
c564f08
9c64f79
fac651e
4e33aee
39848cd
75aeae5
c0c28f9
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,101 @@ | ||
class FibonacciNode: | ||
def __init__(self, key): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: Please provide type hint for the parameter: |
||
self.key = key | ||
self.children = [] | ||
self.marked = False | ||
self.degree = 0 | ||
self.parent = None | ||
|
||
|
||
class FibonacciHeap: | ||
def __init__(self): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. class FibonacciNode: class FibonacciHeap: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: |
||
self.trees = [] | ||
self.least = None | ||
self.count = 0 | ||
|
||
def insert(self, key): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: As there is no test file in this pull request nor any test function or class in the file Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: |
||
new_node = FibonacciNode(key) | ||
self.trees.append(new_node) | ||
self.count += 1 | ||
if self.least is None or key < self.least.key: | ||
self.least = new_node | ||
|
||
def extract_min(self): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: As there is no test file in this pull request nor any test function or class in the file There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: |
||
if self.count == 0: | ||
raise ValueError("Heap is empty") | ||
|
||
min_node = self.least | ||
self.trees.remove(min_node) | ||
|
||
for child in min_node.children: | ||
child.parent = None | ||
self.trees.append(child) | ||
|
||
self.count -= 1 | ||
|
||
if self.least is min_node: | ||
self.least = None | ||
for node in self.trees: | ||
if self.least is None or node.key < self.least.key: | ||
self.least = node | ||
|
||
self.consolidate() | ||
return min_node.key | ||
|
||
def consolidate(self): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: As there is no test file in this pull request nor any test function or class in the file There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: |
||
degree_counts = [None] * (2 * len(self.trees)) | ||
|
||
for node in self.trees[:]: | ||
degree = node.degree | ||
while degree_counts[degree] is not None: | ||
other_node = degree_counts[degree] | ||
if node.key > other_node.key: | ||
node, other_node = other_node, node # Swap nodes | ||
self.link(other_node, node) | ||
degree_counts[degree] = None | ||
degree += 1 | ||
degree_counts[degree] = node | ||
|
||
self.trees = [node for node in degree_counts if node is not None] | ||
|
||
def link(self, node1, node2): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: As there is no test file in this pull request nor any test function or class in the file Please provide type hint for the parameter: Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: Please provide type hint for the parameter: |
||
self.trees.remove(node2) | ||
node1.children.append(node2) | ||
node2.parent = node1 | ||
node1.degree += 1 | ||
node2.marked = False | ||
|
||
def decrease_key(self, node, new_key): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: As there is no test file in this pull request nor any test function or class in the file Please provide type hint for the parameter: Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: Please provide type hint for the parameter: |
||
if new_key > node.key: | ||
raise ValueError("New key must be less than or equal to old key") | ||
|
||
node.key = new_key | ||
|
||
if node == self.least: | ||
self.update_least() | ||
|
||
parent = node.parent | ||
if parent is not None and node.key < parent.key: | ||
self.cut(node, parent) | ||
self.cascading_cut(parent) | ||
|
||
def cut(self, node, parent): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: As there is no test file in this pull request nor any test function or class in the file Please provide type hint for the parameter: Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: Please provide type hint for the parameter: |
||
parent.children.remove(node) | ||
parent.degree -= 1 | ||
self.trees.append(node) | ||
node.parent = None | ||
node.marked = False | ||
|
||
def cascading_cut(self, node): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: As there is no test file in this pull request nor any test function or class in the file Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. class FibonacciNode: class FibonacciHeap:
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: |
||
if (parent := node.parent) is not None: | ||
if not node.marked: | ||
node.marked = True | ||
else: | ||
self.cut(node, parent) | ||
self.cascading_cut(parent) | ||
|
||
def update_least(self): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Please provide return type hint for the function: As there is no test file in this pull request nor any test function or class in the file There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: |
||
self.least = None | ||
for node in self.trees: | ||
if self.least is None or node.key < self.least.key: | ||
self.least = node |
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,28 @@ | ||
import numpy as np | ||
|
||
def metropolis_hastings(target_distribution, proposal_distribution, num_samples, initial_state): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: Please provide type hint for the parameter: Please provide type hint for the parameter: Please provide type hint for the parameter: |
||
samples = [initial_state] | ||
current_state = initial_state | ||
|
||
for _ in range(num_samples): | ||
proposed_state = proposal_distribution(current_state) | ||
acceptance_ratio = min(1, target_distribution(proposed_state) / target_distribution(current_state)) | ||
|
||
if np.random.rand() < acceptance_ratio: | ||
current_state = proposed_state | ||
|
||
samples.append(current_state) | ||
|
||
return samples[1:] | ||
|
||
def target_distribution(x): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: Please provide descriptive name for the parameter: |
||
return np.exp(-x**2 / 2) / np.sqrt(2 * np.pi) | ||
|
||
def proposal_distribution(x): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As there is no test file in this pull request nor any test function or class in the file Please provide return type hint for the function: Please provide type hint for the parameter: Please provide descriptive name for the parameter: |
||
return x + np.random.normal(0, 1) | ||
|
||
if __name__ == "__main__": | ||
num_samples = 10000 | ||
initial_state = 0 | ||
|
||
samples = metropolis_hastings(target_distribution, proposal_distribution, num_samples, initial_state) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please provide return type hint for the function:
__init__
. If the function does not return a value, please provide the type hint as:def function() -> None:
Please provide type hint for the parameter:
key