All examples can use the utility function swap
:
private void swap(int[] a, int i, int j) {
int tmp = a[i];
a[i] = a[j];
a[j] = tmp;
}
public void inOrder(Tree tree) {
if (tree == null) return;
inOrder(tree.left);
// process tree.value
inOrder(tree.right);
}
Without recursion:
public void inOrder(Tree tree) {
Stack<Tree> stack = new Stack<>();
Tree curr = tree;
while (curr != null || !stack.isEmpty()) {
while (curr != null) {
stack.push(curr);
curr = curr.left;
}
curr = stack.pop();
// process curr.value
curr = curr.right;
}
}
public void preOrder(Tree tree) {
if (tree == null) return;
// process tree.value
preOrder(tree.left);
preOrder(tree.right);
}
Without recursion:
public void preOrder2(Tree tree) {
Stack<Tree> stack = new Stack<>();
stack.push(tree);
Tree curr;
while (!stack.isEmpty()) {
curr = stack.pop();
// process curr.value
if (curr.right != null) stack.push(curr.right);
if (curr.left != null) stack.push(curr.left);
}
}
public void postOrder(Tree tree) {
if (tree == null) return;
postOrder(tree.left);
postOrder(tree.right);
// process tree.value
}
Without recursion:
public void postOrder(Tree tree) {
Stack<Tree> tmp = new Stack<>();
Stack<Tree> all = new Stack<>();
tmp.push(tree);
Tree curr;
while (!tmp.isEmpty()) {
curr = tmp.pop();
all.push(curr);
if (curr.left != null) tmp.push(curr.left);
if (curr.right != null) tmp.push(curr.right);
}
while (!all.isEmpty()) {
curr = all.pop();
// process curr.value
}
}
public void sort(int[] a) {
for (int j = 1; j < a.length; j++) {
int key = a[j];
int i = j - 1;
while (i >= 0 && a[i] > key) {
a[i + 1] = a[i];
i--;
}
a[i + 1] = key;
}
}
public void sort(int[] a) {
for (int i = 0; i < a.length; i++)
swap(a, i, min(a, i));
}
private int min(int[] a, int start) {
int smallest = start;
for (int i = start + 1; i < a.length; i++)
if (a[i] < a[smallest])
smallest = i;
return smallest;
}
public int[] sort(int[] a) {
if (a.length <= 1) return a;
return sort(a, 0, a.length - 1);
}
private int[] sort(int[] a, int low, int high) {
if (low == high)
return new int[]{a[low]};
int mid = (low + high) / 2;
int[] sorted1 = sort(a, low, mid);
int[] sorted2 = sort(a, mid + 1, high);
return merge(sorted1, sorted2);
}
private int[] merge(int[] a, int[] b) {
int[] res = new int[a.length + b.length];
int i = 0, j = 0;
while (i < a.length && j < b.length) {
if (a[i] < b[j])
res[i + j] = a[i++];
else
res[i + j] = b[j++];
}
while (i < a.length)
res[i + j] = a[i++];
while (j < b.length)
res[i + j] = b[j++];
return res;
}
public void sort(int[] a) {
sort(a, 0, a.length - 1);
}
private void sort(int[] a, int low, int high) {
if (low >= high) return;
int pivot = partition(a, low, high);
sort(a, low, pivot - 1);
sort(a, pivot + 1, high);
}
private int partition(int[] a, int low, int high) {
int pivot = low;
int rand = new Random().nextInt(high - low + 1) + low;
swap(a, low, rand);
for (int i = low + 1; i <= high; i++) {
if (a[i] < a[pivot]) {
swap(a, i, pivot + 1);
swap(a, pivot, pivot + 1);
pivot++;
}
}
return pivot;
}
public void sort(int[] a) {
// build a max-heap
for (int i = a.length - 1; i >= 0; i--)
heapify(a, i, a.length);
// extract max element from the head to the end and shrink the size of the heap
for (int last = a.length - 1; last >= 0; last--) {
swap(a, 0, last);
heapify(a, 0, last);
}
}
// heapify for a max-heap:
private void heapify(int[] a, int root, int length) {
int left = 2 * root + 1;
int right = 2 * root + 2;
int largest = root;
if (left < length && a[largest] < a[left])
largest = left;
if (right < length && a[largest] < a[right])
largest = right;
if (largest != root) {
swap(a, root, largest);
heapify(a, largest, length);
}
}
public int[] sort(int[] a) {
int max = findMax(a);
int[] sorted = new int[a.length];
int[] counts = new int[max + 1];
for (int i = 0; i < a.length; i++)
counts[a[i]]++;
for (int i = 1; i < counts.length; i++)
counts[i] += counts[i - 1];
for (int i = 0; i < a.length; i++) {
sorted[counts[a[i]] - 1] = a[i];
counts[a[i]]--;
}
return sorted;
}
private int findMax(int[] a) {
if (a.length == 0) return 0;
int max = Integer.MIN_VALUE;
for (int i = 0; i < a.length; i++) {
if (a[i] > max)
max = a[i];
}
return max;
}
public int search(int[] a, int x) {
return search(a, x, 0, a.length - 1);
}
private int search(int[] a, int x, int low, int high) {
if (low > high) return -1;
int mid = (low + high) / 2;
if (a[mid] == x) return mid;
else if (a[mid] > x) return search(a, x, low, mid - 1);
else return search(a, x, mid + 1, high);
}
public int quickselect(int[] a, int k) {
return quickselect(a, k, 0, a.length - 1);
}
private int quickselect(int[] a, int k, int low, int high) {
int pivot = partition(a, low, high);
if (pivot == k) return a[pivot];
else if (k < pivot) return quickselect(a, k, low, pivot - 1);
else return quickselect(a, k, pivot + 1, high);
}
private int partition(int[] a, int low, int high) {
int pivot = low;
int rand = new Random().nextInt(high - low + 1) + low;
swap(a, low, rand);
for (int i = low + 1; i <= high; i++) {
if (a[i] < a[pivot]) {
swap(a, i, pivot + 1);
swap(a, pivot, pivot + 1);
pivot++;
}
}
return pivot;
}
Utility function:
private void reset(Graph graph) {
for (Vertex v : graph.vertices.values()) {
v.parent = null;
v.discovered = false;
v.distance = Integer.MAX_VALUE;
}
}
public void bfs(Graph graph, Vertex source) {
reset(graph);
Queue<Vertex> q = new LinkedList<>();
q.add(source);
source.discovered = true;
while (!q.isEmpty()) {
Vertex from = q.remove();
for (Edge e : from.edges) {
Vertex to = graph.vertices.get(e.to);
if (!to.discovered) {
to.parent = from;
to.discovered = true;
q.add(to);
}
}
}
}
public void dfs(Graph graph) {
reset(graph);
for (Vertex v : graph.vertices.values()) {
if (!v.discovered)
dfs(graph, v);
}
}
private void dfs(Graph graph, Vertex v) {
v.discovered = true;
// TODO: insert application of DFS here
for (Edge e : v.edges) {
Vertex to = graph.vertices.get(e.to);
if (to.discovered) {
// cycle found (back edge)! What should we do? depends on the application...
} else {
to.parent = v;
dfs(graph, to);
}
}
// TODO: insert application of DFS here. For example: if we're doing topological sorting
// then add v to head of a linked list at this point
}
public void dijkstra(Graph graph, Vertex source) {
reset(graph);
source.distance = 0;
PriorityQueue<Vertex> q = new PriorityQueue<>(graph.vertices.values());
while (!q.isEmpty()) {
Vertex from = q.remove();
for (Edge edge : from.edges) {
Vertex to = graph.vertices.get(edge.to);
int newDistance = from.distance + edge.weight;
if (newDistance < to.distance) {
to.distance = newDistance;
to.parent = from;
q.remove(to);
q.add(to);
}
}
}
}