Algorithms R OBERT S EDGEWICK | K EVIN W AYNE

2.4 P RIORITY Q UEUES

‣ API and elementary implementations
‣ binary heaps
‣ heapsort
Algorithms
‣ event-driven simulation
F O U R T H E D I T I O N

R OBERT S EDGEWICK | K EVIN W AYNE

http://algs4.cs.princeton.edu
2.4 P RIORITY Q UEUES
‣ API and elementary implementations
‣ binary heaps
‣ heapsort
Algorithms
‣ event-driven simulation

R OBERT S EDGEWICK | K EVIN W AYNE

http://algs4.cs.princeton.edu
Collections

A collection is a data type that stores a group of items.

data type core operations data structure

stack PUSH, POP linked list, resizing array

queue ENQUEUE, DEQUEUE linked list, resizing array

priority queue INSERT, DELETE-MAX binary heap

symbol table PUT, GET, DELETE binary search tree, hash table

set ADD, CONTAINS, DELETE binary search tree, hash table

“ Show me your code and conceal your data structures, and I shall
continue to be mystified. Show me your data structures, and I won't
usually need your code; it'll be obvious.” — Fred Brooks

3
Priority queue

Collections. Insert and delete items. Which item to delete?

Stack. Remove the item most recently added.

Queue. Remove the item least recently added.
Randomized queue. Remove a random item.

Priority queue. Remove the largest (or smallest) item.

Generalizes: stack, queue, randomized queue. operation argument
return co
value size (un

insert P 1 P
insert Q 2 P Q
insert E 3 P Q
remove max Q 2 P E
insert X 3 P E
insert A 4 P E
insert M 5 P E
remove max X 4 P E
insert P 5 P E
insert L 6 P E
insert E 7 P E
remove max P 6 E M
4
A sequence of op
Priority queue API

Requirement. Items are generic; they must also be Comparable.

Key must be Comparable

(bounded type parameter)

public class MaxPQ<Key extends Comparable<Key>>

MaxPQ() create an empty priority queue

MaxPQ(Key[] a) create a priority queue with given keys

void insert(Key v) insert a key into the priority queue

Key delMax() return and remove a largest key

boolean isEmpty() is the priority queue empty?

Key max() return a largest key

int size() number of entries in the priority queue

Note. Duplicate keys allowed; delMax() picks any maximum key.

5
Priority queue: applications

・Event-driven simulation. [ customers in a line, colliding particles ]

・Numerical computation. [ reducing roundoff error ]
・Discrete optimization. [ bin packing, scheduling ]
・Artificial intelligence. [ A* search ]
・Computer networks. [ web cache ]
・Operating systems. [ load balancing, interrupt handling ]
・Data compression. [ Huffman codes ]
・Graph searching. [ Dijkstra's algorithm, Prim's algorithm ]
・Number theory. [ sum of powers ]
・Spam filtering. [ Bayesian spam filter ]
・Statistics. [ online median in data stream ]

6
Priority queue: client example

Challenge. Find the largest M items in a stream of N items.

・Fraud detection: isolate $$ transactions.
・NSA monitoring: flag most suspicious documents. N huge, M large

Constraint. Not enough memory to store N items.

Transaction data
type is Comparable
(ordered by $$)

MinPQ<Transaction> pq = new MinPQ<Transaction>();

while (StdIn.hasNextLine())
use a min-oriented pq {
String line = StdIn.readLine();
Transaction transaction = new Transaction(line);
pq.insert(transaction);
if (pq.size() > M)
pq now contains
pq.delMin(); largest M items
}

7
Priority queue: client example

Challenge. Find the largest M items in a stream of N items.

implementation time space

sort N log N N

elementary PQ MN M

binary heap N log M M

best in theory N M

order of growth of finding the largest M in a stream of N items

8
Priority queue: unordered and ordered array implementation

return contents contents

operation argument value size (unordered) (ordered)

insert P 1 P P
insert Q 2 P Q P Q
insert E 3 P Q E E P Q
remove max Q 2 P E E P
insert X 3 P E X E P X
insert A 4 P E X A A E P X
insert M 5 P E X A M A E M P X
remove max X 4 P E M A A E M P
insert P 5 P E M A P A E M P P
insert L 6 P E M A P L A E L M P P
insert E 7 P E M A P L E A E E L M P P
remove max P 6 E M A P L E A E E L M P

A sequence of operations on a priority queue

9
Priority queue: implementations cost summary

Challenge. Implement all operations efficiently.

implementation insert del max max

unordered array 1 N N

ordered array N 1 1

goal log N log N log N

order of growth of running time for priority queue with N items

10
2.4 P RIORITY Q UEUES
‣ API and elementary implementations
‣ binary heaps
‣ heapsort
Algorithms
‣ event-driven simulation

R OBERT S EDGEWICK | K EVIN W AYNE

http://algs4.cs.princeton.edu
Complete binary tree

Binary tree. Empty or node with links to left and right binary trees.

Complete tree. Perfectly balanced, except for bottom level.

complete binary tree with N = 16 nodes (height = 4)

Property. Height of complete binary tree with N nodes is !lg N".

Pf. Height increases only when N is a power of 2.

12
A complete binary tree in nature

13
Binary heap: representation

Binary heap. Array representation of a heap-ordered complete binary tree.

Heap-ordered binary tree.

・Keys in nodes.
・Parent's key no smaller than
children's keys.
i 0 1 2 3 4 5 6 7 8 9 10 11
a[i] - T S R P N O A E I H G

Array representation. T

・Indices start at 1.
S R

・Take nodes in level order.

P N O A

・No explicit links needed! E I H G

1
T
2 S 3 R

4 P 5 N 6 O 7 A

8 E 9 I 10 H 11 G

Heap representations

14
Binary heap: properties

Proposition. Largest key is a[1], which is root of binary tree.

Proposition. Can use array indices to move through tree.

・Parent of node at k is at k/2.
・Children of node at k are at 2k and 2k+1.
i 0 1 2 3 4 5 6 7 8 9 10 11
a[i] - T S R P N O A E I H G

T
S R

P N O A

E I H G
1
T
2 S 3 R

4 P 5 N 6 O 7 A

8 E 9 I 10 H 11 G

Heap representations

15
Binary heap demo

Insert. Add node at end, then swim it up.

Remove the maximum. Exchange root with node at end, then sink it down.

heap ordered

P R

N H O A

E I G

T P R N H O A E I G

16
Binary heap demo

Insert. Add node at end, then swim it up.

Remove the maximum. Exchange root with node at end, then sink it down.

heap ordered

R O

N P G A

E I H

S R O N P G A E I H

17
Binary heap: promotion

Scenario. A key becomes larger than its parent's key.

To eliminate the violation:

・Exchange key in child with key in parent.
・Repeat until heap order restored. S

private void swim(int k) P R

{ N 5 T O A
while (k > 1 && less(k/2, k)) E I H G
violates heap order
(larger key than parent)
{
exch(k, k/2); 1
T
k = k/2;
2
S R
} parent of node at k is at k/2
} N 5 P O A

E I H G

Bottom-up reheapify (swim)

Peter principle. Node promoted to level of incompetence.

18
Binary heap: insertion

Insert. Add node at end, then swim it up.

Cost. At most 1 + lg N compares.
insert remo
T

P R

N H O A

public void insert(Key x) E I G S key to insert

{
T
pq[++N] = x;
swim(N); P R

} N H O A

E I G S add key to heap

violates heap order

T
swim up
S R
sink down
N P O A

E I G H

Heap operation

19
Binary heap: demotion

Scenario. A key becomes smaller than one (or both) of its children's.

why not smaller child?

To eliminate the violation:
・Exchange key in parent with key in larger child.
・Repeat until heap order restored.
private void sink(int k) violates heap order
(smaller than a child)
T
{ 2
children of node at k H R
while (2*k <= N)
are 2*k and 2*k+1 5 S
{ P O A

int j = 2*k; E I N G

if (j < N && less(j, j+1)) j++; T

if (!less(k, j)) break; 2
S R
exch(k, j); 5
P N O A
k = j; 10
E I H G
}
} Top-down reheapify (sink)

Power struggle. Better subordinate promoted.

20
Binary heap: delete the maximum

Delete max. Exchange root with node at end, then sink it down.
Cost. At most 2 lg N compares.

insert remove the maximum

T T key to remove

P R S R

N H O A N P O A
public Key delMax()
exchange key
{ E I G S key to insert E I G H with root
Key max = pq[1];
violates
exch(1, N--); T H heap order
sink(1); P R S R
pq[N+1] = null;
N H
prevent loitering
O A N P O A
return max; remove node
E I G S add key to heap E I G T
} violates heap order from heap

T S
swim up
S R P R
sink down
N P O A N H O A

E I G H E I G

Heap operations
21
Binary heap: Java implementation

public class MaxPQ<Key extends Comparable<Key>>

{
private Key[] pq;
private int N;

public MaxPQ(int capacity) fixed capacity

(for simplicity)
{ pq = (Key[]) new Comparable[capacity+1]; }

public boolean isEmpty() PQ ops

{ return N == 0; }
public void insert(Key key) // see previous code
public Key delMax() // see previous code

private void swim(int k) // see previous code

heap helper functions
private void sink(int k) // see previous code

private boolean less(int i, int j)

{ return pq[i].compareTo(pq[j]) < 0; }
array helper functions
private void exch(int i, int j)
{ Key t = pq[i]; pq[i] = pq[j]; pq[j] = t; }
}

22
Priority queue: implementations cost summary

implementation insert del max max

unordered array 1 N N

ordered array N 1 1

binary heap log N log N 1

order-of-growth of running time for priority queue with N items

23
DELETE-RANDOM FROM A BINARY HEAP

Goal. Delete a random key from a binary heap in logarithmic time.

R H

P N G A

E J I

24
Binary heap: practical improvements

Do "half-exchanges" in sink and swim.

・Reduces number of array accesses.
・Worth doing.

27
Binary heap: practical improvements

Floyd's "bounce" heuristic.

・Sink key at root all the way to bottom. only 1 compare per node

・Swim key back up. some extra compares and exchanges

・Overall, fewer compares; more exchanges.

・Worthwhile depending on cost of compare and exchange. R. W. Floyd
1978 Turing award

X Y

N O

L K

E D

28
Binary heap: practical improvements

Multiway heaps.
・Complete d-way tree.
・Parent's key no smaller than its children's keys.
Fact. Height of complete d-way tree on N nodes is ~ logd N.

Y T G

J X P W I K A B D

E H F R S V C L M Q N O

3-way heap

29
Priority queues: quiz 1

How many compares (in the worst case) to insert in a d-way heap?

A. ~ log2 N

B. ~ logd N

C. ~ d log2 N

D. ~ d logd N

E. I don't know.

30
Priority queues: quiz 2

How many compares (in the worst case) to delete-max in a d-way heap?

A. ~ log2 N

B. ~ logd N

C. ~ d log2 N

D. ~ d logd N

E. I don't know.

31
Priority queue: implementation cost summary

implementation insert del max max

unordered array 1 N N

ordered array N 1 1

binary heap log N log N 1

d-ary heap logd N d logd N 1 sweet spot: d = 4

Fibonacci 1 log N † 1

Brodal queue 1 log N 1

impossible 1 1 1 why impossible?

† amortized

order-of-growth of running time for priority queue with N items

32
Binary heap: considerations

Underflow and overflow.

・Underflow: throw exception if deleting from empty PQ.
・Overflow: add no-arg constructor and use resizing array.
leads to log N
amortized time per op
Minimum-oriented priority queue. (how to make worst case?)

・Replace less() with greater().

・Implement greater().
Other operations.
・Remove an arbitrary item. can implement efficiently with sink() and swim()

・Change the priority of an item.

[ stay tuned for Prim/Dijkstra ]

Immutability of keys.
・Assumption: client does not change keys while they're on the PQ.
・Best practice: use immutable keys.
33
Immutability: implementing in Java

Data type. Set of values and operations on those values.

Immutable data type. Can't change the data type value once created.

public class Vector {

instance variables private and final
private final int N;
(neither necessary nor sufficient,
private final double[] data; but good programming practice)

public Vector(double[] data) {

this.N = data.length;
this.data = new double[N];
defensive copy of mutable
for (int i = 0; i < N; i++) instance variables
this.data[i] = data[i];
}
instance methods don't
⋮ change instance variables
}

Immutable. String, Integer, Double, Color, Vector, Transaction, Point2D.

Mutable. StringBuilder, Stack, Counter, Java array.
34
Immutability: properties

Data type. Set of values and operations on those values.

Immutable data type. Can't change the data type value once created.

Advantages.
・Simplifies debugging.
・Simplifies concurrent programming.
・More secure in presence of hostile code.
・Safe to use as key in priority queue or symbol table.
Disadvantage. Must create new object for each data type value.

“ Classes should be immutable unless there's a very good reason

to make them mutable.… If a class cannot be made immutable,
you should still limit its mutability as much as possible. ”
— Joshua Bloch (Java architect)

35
2.4 P RIORITY Q UEUES
‣ API and elementary implementations
‣ binary heaps
‣ heapsort
Algorithms
‣ event-driven simulation

R OBERT S EDGEWICK | K EVIN W AYNE

http://algs4.cs.princeton.edu
Priority queues: quiz 3

What is the name of this sorting algorithm?

public void sort(String[] a)

{
int N = a.length;
MaxPQ<String> pq = new MaxPQ<String>();
for (int i = 0; i < N; i++)
pq.insert(a[i]);
for (int i = N-1; i >= 0; i--)
a[i] = pq.delMax();
}

A. Insertion sort.

B. Mergesort.

C. Quicksort.

D. None of the above.

E. I don't know.

37
Priority queues: quiz 4

What are its properties?

public void sort(String[] a)

{
int N = a.length;
MaxPQ<String> pq = new MaxPQ<String>();
for (int i = 0; i < N; i++)
pq.insert(a[i]);
for (int i = N-1; i >= 0; i--)
a[i] = pq.delMax();
}

A. N log N compares in the worst case.

B. In-place.

C. Stable.

D. All of the above.

E. I don't know.

38
k(3, 11) exch(1, 9) exch(1, 3)
Heapsort
S
sink(1, 8)
R
sink(1, 2)
E
O X P E A E

T L
Basic plan
R for
A in-place sort. O L E A L M O P

・EView input array as a complete

P E M S binary
T X tree. R S T X

・ Heap construction: build a max-heap with all N keys.

・Sortdown: repeatedly
k(2, 11) S exch(1, 8) P exch(1, 2) A
remove
sink(1, 7) the maximum key.sink(1, 1)
T X O E E E

P L R A M L E A L M O P
O E E R S T X R S T X
keys in arbitrary order build max heap sorted result
heap construction
heap construction sortdownsortdown
(in place) (in place)

1
S 1
S exch(1, 7) XO X exch(1,exch(1,
6) 6) M1 A M
sink(1, 6) sink(1,sink(1,
5) 5)
k(1, 211) 2 XO 3 3
O R R T
M T SE S L2 E L E3 E E

TT S
4 4 5 56 7 6 7
T E E
X A X A AP P L
L LRE APR A A4 L A E5 M EO 6 O P O7 P P
9 8 10 9 11 10
P M LL P E R
11
PM L EA M O E E E E R8 S9 R T10S X11T X
R S M T O X R S T X
starting point (arbitrary
starting order) order) starting point (heap-ordered)
starting point (heap-ordered)
O E E point (arbitrary result (sorted)
result (heap-ordered)
sink(5, sink(5, 11)
11) 1 2 3 4 S5 6 7 S8 exch(1,
9 1 2exch(1,
11)
10
3 411511)
6 7T 8 9 T 1 exch(1,
10 11 exch(1, 2 5)3 4 5)
5 6 L
7 8 9 L 10 11

XHeapsort:
A M P constructing
X (left)
T S andP sorting
L R A down
M O (right)
E E a heap
sink(1, sink(1,
10) 10) sink(1,sink(1,
4) 4)
S O R T E L E A E E L M O P R S T X
O O R R P P S S E E E E

T T L L
X A X A O O L LR A R A A A M MO PO P

M P M E P E E E M E M E E X E X R SR TS XT X
39
Heapsort demo

Heap construction. Build max heap using bottom-up method.

we assume array entries are indexed 1 to N

array in arbitrary order

1 S

2 O 3 R

4 T 5 E 6 X 7 A

8 M 9 P 10 L 11 E

S O R T E X A M P L E
1 2 3 4 5 6 7 8 9 10 11
40
Heapsort demo

Sortdown. Repeatedly delete the largest remaining item.

array in sorted order

E E

L M O P

R S T X

A E E L M O P R S T X
1 2 3 4 5 6 7 8 9 10 11
41
sink(5, 11) S exch(1, 11) T
sink(1, 10)
Heapsort: heap construction O R P

T L X A O L R
M P E E M E E X
First pass. Build heap using bottom-up method.
sink(4, 11) S exch(1, 10) S
sink(1, 9)
O R P
for (int k = N/2; k >= 1; k--)
T L X A O L E
sink(a, k, N);
M P E E M E T X

heap construction sortdown

sink(3, 11) S exch(1, 9) R
1
S X exch(1,sink(1,
6) 8)
M
O X sink(1, 5) P
2 3
O R T S L E
T L R A O L E
4 5 6 7
T E X A P L R A A E O P
8 9 10 11 M P E E M S T X
M P L E M O E E R S T X
starting point (arbitrary order) starting point (heap-ordered)
sink(2, 11) S exch(1, 8) P
sink(1, 7)
sink(5, 11) S exch(1, 11)T T X exch(1, 5) O L
sink(1, 10) sink(1, 4)
O R P P L R SA E M L E E
T L X A M OO E LE R A A R SM T O X P
M P E E M E E X R S T X
exch(1, 7) O
sink(1, 11) sink(1, 6)
sink(4, 11) S exch(1, 10) XS exch(1, 4)
sink(1, 9) sink(1, 3)
M E
O R TP SR A A L E E
L PO LL RE AA
T X A L R SM T O X P
P E E MM OE ET EX S T X
M R
result (heap-ordered)
Heapsort: constructing (left) and sorting down
42
Heapsort: sortdown
heap
heap construction
construction sortdown
sortdown

Second pass. 11
S X exch(1,
exch(1, 6)
6) M

・Remove the maximum, one at a time.

sink(1,
sink(1, 5)
5)
22 33
O R T S L E
44 55 66 77
T E X A P L R A A E O P

・Leave in array, instead of nulling out.

88 99 10
10 11
11
M P L E M O E E R S T X
starting point (arbitrary order) starting point (heap-ordered)

sink(5,
sink(5, 11)
11) S exch(1,
exch(1, 11)
11) T exch(1,
exch(1, 5)
5) L
sink(1,
sink(1, 10)
10) sink(1,
sink(1, 4)
4)
O R P S E E
L A L A M P
while (N > 1) T X O R A O
M P E E M E E X R S T X
{
exch(a, 1, N--); sink(4,
sink(4, 11)
11) S exch(1,
exch(1, 10)
sink(1,
10)
sink(1, 9)
9)
S exch(1,
exch(1, 4)
sink(1,
4)
sink(1, 3)
3)
E
O R P R A E
sink(a, 1, N); L X A L E A M O P
T O L
} M P E E M E T X R S T X

sink(3,
sink(3, 11)
11) S exch(1,
exch(1, 9)
9) R exch(1,
exch(1, 3)
3) E
sink(1,
sink(1, 8)
8) sink(1,
sink(1, 2)
2)
O X P E A E

T L R A O L E A L M O P
M P E E M S T X R S T X

sink(2,
sink(2, 11)
11) S exch(1,
exch(1, 8)
8) P exch(1,
exch(1, 2)
2) A
sink(1,
sink(1, 7)
7) sink(1,
sink(1, 1)
1)
T X O E E E

P L R A M L E A L M O P
M O E E R S T X R S T X

exch(1,
exch(1, 7)
7) O 11
A
sink(1,
sink(1, 11)
11) sink(1,
sink(1, 6)
6)
X 22 33
M E E E
T S 44 55 66 77
A L E P L M O P
P L R A 88 99 10
10 11
11
R S T X R S T X
M O E E result (sorted)
result (heap-ordered) 43
Heapsort: constructing (left) and sorting down (right) a heap
Heapsort: Java implementation

public class Heap

{
public static void sort(Comparable[] a)
{
int N = a.length;
for (int k = N/2; k >= 1; k--)
sink(a, k, N);
while (N > 1)
{
exch(a, 1, N);
sink(a, 1, --N);
}
}
but make static (and pass arguments)

private static void sink(Comparable[] a, int k, int N)

{ /* as before */ }

private static boolean less(Comparable[] a, int i, int j)

{ /* as before */ }

private static void exch(Object[] a, int i, int j)

{ /* as before */ }
but convert from 1-based
} indexing to 0-base indexing

44
Heapsort: trace

a[i]
N k 0 1 2 3 4 5 6 7 8 9 10 11
initial values S O R T E X A M P L E
11 5 S O R T L X A M P E E
11 4 S O R T L X A M P E E
11 3 S O X T L R A M P E E
11 2 S T X P L R A M O E E
11 1 X T S P L R A M O E E
heap-ordered X T S P L R A M O E E
10 1 T P S O L R A M E E X
9 1 S P R O L E A M E T X
8 1 R P E O L E A M S T X
7 1 P O E M L E A R S T X
6 1 O M E A L E P R S T X
5 1 M L E A E O P R S T X
4 1 L E E A M O P R S T X
3 1 E A E L M O P R S T X
2 1 E A E L M O P R S T X
1 1 A E E L M O P R S T X
sorted result A E E L M O P R S T X

Heapsort trace (array contents just after each sink)

45
Heapsort: mathematical analysis

Proposition. Heap construction makes ≤ N exchanges and ≤ 2 N compares.

Pf sketch. [assume N = 2h+1 – 1]

max number of exchanges

to sink node
3
3

2 2
2 2
1 1 1 1
1 1 1 1

0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0

binary heap of height h = 3

a tricky sum
(see COS 340)

h + 2(h 1) + 4(h 2) + 8(h 3) + . . . + 2h (0) 22h+1

h+1
1
= NN
46
Heapsort: mathematical analysis

Proposition. Heap construction uses ≤ 2 N compares and ≤ N exchanges.

Proposition. Heapsort uses ≤ 2 N lg N compares and exchanges.

algorithm can be improved to ~ 1 N lg N

(but no such variant is known to be practical)

Significance. In-place sorting algorithm with N log N worst-case.

・Mergesort: no, linear extra space. in-place merge possible, not practical

・Quicksort: no, quadratic time in worst case. N log N worst-case quicksort possible,

・Heapsort: yes!
not practical

Bottom line. Heapsort is optimal for both time and space, but:
・Inner loop longer than quicksort’s.
・Makes poor use of cache.
・Not stable. can be improved using
advanced caching tricks

47
Introsort

Goal. As fast as quicksort in practice; N log N worst case, in place.

Introsort.
・Run quicksort.
・Cutoff to heapsort if stack depth exceeds 2 lg N.
・Cutoff to insertion sort for N = 16.

In the wild. C++ STL, Microsoft .NET Framework.

48
Sorting algorithms: summary

inplace? stable? best average worst remarks

selection ✔ ½N2 ½N2 ½N2 N exchanges

use for small N

insertion ✔ ✔ N ¼N2 ½N2
or partially ordered
tight code;
shell ✔ N log3 N ? c N 3/2
subquadratic

N log N guarantee;
merge ✔ ½ N lg N N lg N N lg N
stable
improves mergesort
timsort ✔ N N lg N N lg N
when preexisting order

quick
N log N probabilistic guarantee;
✔ N lg N 2 N ln N ½N2
fastest in practice
improves quicksort
3-way quick ✔ N 2 N ln N ½N2
when duplicate keys

heap N log N guarantee;

✔ N 2 N lg N 2 N lg N
in-place

? ✔ ✔ N N lg N N lg N holy sorting grail

49
2.4 P RIORITY Q UEUES
‣ API and elementary implementations
‣ binary heaps
‣ heapsort
Algorithms
‣ event-driven simulation

R OBERT S EDGEWICK | K EVIN W AYNE

http://algs4.cs.princeton.edu
Molecular dynamics simulation of hard discs

Goal. Simulate the motion of N moving particles that behave

according to the laws of elastic collision.

51
Molecular dynamics simulation of hard discs

Goal. Simulate the motion of N moving particles that behave

according to the laws of elastic collision.

Hard disc model.

・Moving particles interact via elastic collisions with each other and walls.
・Each particle is a disc with known position, velocity, mass, and radius.
・No other forces.

temperature, pressure, motion of individual

diffusion constant atoms and molecules

Significance. Relates macroscopic observables to microscopic dynamics.

・Maxwell-Boltzmann: distribution of speeds as a function of temperature.
・Einstein: explain Brownian motion of pollen grains.

52
Warmup: bouncing balls

Time-driven simulation. N bouncing balls in the unit square.

public class BouncingBalls % java BouncingBalls 100

{
public static void main(String[] args)
{
int N = Integer.parseInt(args[0]);
Ball[] balls = new Ball[N];
for (int i = 0; i < N; i++)
balls[i] = new Ball();
while(true)
{
StdDraw.clear();
for (int i = 0; i < N; i++)
{
balls[i].move(0.5);
balls[i].draw();
}
StdDraw.show(50);
}
} main simulation loop

53
Warmup: bouncing balls

public class Ball

{
private double rx, ry; // position
private double vx, vy; // velocity
private final double radius; // radius
public Ball(...)
{ /* initialize position and velocity */ } check for collision with walls

public void move(double dt)

{
if ((rx + vx*dt < radius) || (rx + vx*dt > 1.0 - radius)) { vx = -vx; }
if ((ry + vy*dt < radius) || (ry + vy*dt > 1.0 - radius)) { vy = -vy; }
rx = rx + vx*dt;
ry = ry + vy*dt;
}
public void draw()
{ StdDraw.filledCircle(rx, ry, radius); }
}

Missing. Check for balls colliding with each other.

・Physics problems: when? what effect?
・CS problems: which object does the check? too many checks?
54
Time-driven simulation

・Discretize time in quanta of size dt.

・Update the position of each particle after every dt units of time,
and check for overlaps.
・If overlap, roll back the clock to the time of the collision, update the
velocities of the colliding particles, and continue the simulation.

t t + dt t + 2 dt t + Δt
(collision detected) (roll back clock)

55
Time-driven simulation

Main drawbacks.
・~ N / 2 overlap checks per time quantum.
2

・Simulation is too slow if dt is very small.

・ May miss collisions if dt is too large. dt too small: excessive computation

(if colliding particles fail to overlap when we are looking)

dt too small: excessive computation dt too large: may miss collisions

dt too large: may miss collisions

Fundamental challenge for
time-driven simulation

56
Event-driven simulation

Change state only when something interesting happens.

・Between collisions, particles move in straight-line trajectories.
・Focus only on times when collisions occur.
・Maintain PQ of collision events, prioritized by time.
・Delete min = get next collision.
Collision prediction. Given position, velocity, and radius of a particle,
when will it collide next with a wall or another particle?

Collision resolution. If collision occurs, update colliding particle(s)

according to laws of elastic collisions.

prediction (at time t)

particles hit unless one passes
intersection point before the other
arrives (see Exercise 3.6.X)

resolution (at time t + dt)

velocities of both particles
change after collision (see Exercise 3.6.X)

Predicting and resolving a particle-particle collision 57

Particle-wall collision

Collision prediction and resolution.

・Particle of radius s at position (rx, ry).
・Particle moving in unit box with velocity (vx, vy).
・Will it collide with a vertical wall? If so, when?

s
resolution (at time t + dt)
velocity after collision = ( − vx , vy)
position after collision = ( 1 − s , ry + vydt)
prediction (at time t)
dt ! time to hit wall
wall at
= distance/velocity (rx , ry )
x=1
= (1 − s − rx )/vx vy
vx
1 − s − rx

Predicting and resolving a particle-wall collision

58
Particle-particle collision prediction

Collision prediction.
・Particle i: radius s , position (rx , ry ), velocity (vx , vy ).
i i i i i

・Particle j: radius s , position (rx , ry ), velocity (vx , vy ).

j j j j j

・Will particles i and j collide? If so, when?

(vxi', vyi')

(vxj', vyj')
mi

si (vxi , vyi )

(rxi , ryi)
(rxi', ryi')

i
time = t time = t + Δt

(vxj , vyj)

σj

59
Particle-particle collision prediction

Collision prediction.
・Particle i: radius s , position (rx , ry ), velocity (vx , vy ).
i i i i i

・Particle j: radius s , position (rx , ry ), velocity (vx , vy ).

j j j j j

・Will particles i and j collide? If so, when?

B7 v · r 0,
t= B7 d < 0,
v· r + d
Qi?2`rBb2
v· v

d = (∆v · ∆r)2 − (∆v · ∆v) (∆r · ∆r − s2 ), s = si + sj

Δv = (Δvx, Δvy) = (vxi − vx j , vyi − vy j ) Δv ⋅ Δv = (Δvx)2 + (Δvy)2

Δr = (Δrx, Δry) = (rxi − rx j , ryi − ry j ) Δr ⋅ Δr = (Δrx)2 + (Δry)2
Δv ⋅ Δr = (Δvx)(Δrx) + (Δvy)(Δry)

€ €
Important note: This is physics, so we won’t be testing you on it!
€ € 60
Particle-particle collision resolution

Collision resolution. When two particles collide, how does velocity change?

vxi" vxii" + =Jx vx

= / mi i + Jx / mi
vyi" vyii" + =Jy /vymi i + Jy / mi
= vy Newton's second law
(momentum form)
vx j" = vx j" − =Jx vx
/ mj j− Jx / m j

j vx j" − =Jy vx
vy " = vy j / mj − Jy / m j
j

€ €

J rx J ry 2 mi mj ( v · r)
Jx = , Jy = , J=
s s s (mi + mj )
impulse due to normal force
(conservation of energy, conservation of momentum)

Important note: This is physics, so we won’t be testing you on it!

61
Particle data type skeleton

public class Particle

{
private double rx, ry; // position
private double vx, vy; // velocity
private final double radius; // radius
private final double mass; // mass
private int count; // number of collisions

public Particle( ... ) { ... }

public void move(double dt) { ... }

public void draw() { ... }

public double timeToHit(Particle that) { }

predict collision
public double timeToHitVerticalWall() { } with particle or wall
public double timeToHitHorizontalWall() { }

public void bounceOff(Particle that) { }

resolve collision
public void bounceOffVerticalWall() { } with particle or wall
public void bounceOffHorizontalWall() { }

http://algs4.cs.princeton.edu/61event/Particle.java.html
62
Collision system: event-driven simulation main loop

Initialization. two particles on a collision course

・Fill PQ with all potential particle-wall collisions.

・Fill PQ with all potential particle-particle collisions.
third particle interferes: no collision
“potential” since collision is invalidated
if some other collision intervenes

An invalidated event

Main loop.
・Delete the impending event from PQ (min priority = t).
・If the event has been invalidated, ignore it.
・Advance all particles to time t, on a straight-line trajectory.
・Update the velocities of the colliding particle(s).
・Predict future particle-wall and particle-particle collisions involving the
colliding particle(s) and insert events onto PQ.

63
Event data type

Conventions.

・Neither particle null ⇒ particle-particle collision.

・One particle null ⇒ particle-wall collision.

・Both particles null ⇒ redraw event.

private static class Event implements Comparable<Event>

{
private final double time; // time of event
private final Particle a, b; // particles involved in event
private final int countA, countB; // collision counts of a and b

public Event(double t, Particle a, Particle b) create event

{ ... }

public int compareTo(Event that) ordered by time

{ return this.time - that.time; }

public boolean isValid() valid if no intervening collisions

{ ... } (compare collision counts)

64
Particle collision simulation: example 1

% java CollisionSystem 100

65
Particle collision simulation: example 2

% java CollisionSystem < billiards.txt

66
Particle collision simulation: example 3

% java CollisionSystem < brownian.txt

67
Particle collision simulation: example 4

% java CollisionSystem < diffusion.txt

68