X-Git-Url: http://git.megacz.com/?p=sbp.git;a=blobdiff_plain;f=src%2Fedu%2Fberkeley%2Fsbp%2Futil%2FFastSet.java;h=289e5ec22c1546ae98569aff99368a5aa64164d4;hp=581886ec0163f3508ad16bbc3acf3de8648291a9;hb=f069d11a0bc59d63b078df8a4aa488498c4e9cc2;hpb=476b0f25eb4554febad2bbed314bb48a888400c5

diff --git a/src/edu/berkeley/sbp/util/FastSet.java b/src/edu/berkeley/sbp/util/FastSet.java
index 581886e..289e5ec 100644
--- a/src/edu/berkeley/sbp/util/FastSet.java
+++ b/src/edu/berkeley/sbp/util/FastSet.java
@@ -3,7 +3,7 @@
 package edu.berkeley.sbp.util;
 import java.util.*;
 
-public /*final*/ class FastSet<T> implements Iterator<T>, Iterable<T>, Visitable<T> {
+public final class FastSet<T> implements Iterator<T>, Iterable<T>, Visitable<T> {
 
     public static final int INITIAL_SIZE = 8;
 
@@ -42,11 +42,16 @@ public /*final*/ class FastSet<T> implements Iterator<T>, Iterable<T>, Visitable
             return;
         }
         boolean found = false;
-        for(int j=0; j<size; j++) {
-            if (array[j]==t) found = true;
-            if (found && j<size-1) array[j] = array[j+1];
-        }
-        if (found) size--;
+        int j;
+        for(j=0; j<size; j++)
+            if (array[j]==t) {
+                found = true;
+                break;
+            }
+        if (!found) return;
+        array[j] = array[size-1];
+        array[size-1] = null;
+        size--;
     }
 
     public <B,C> void visit(Invokable<T,B,C> ivbc, B b, C c) {