X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=src%2Fedu%2Fberkeley%2Fsbp%2FResult.java;h=10a9b9434b3b81bbe52ea7c8fefe367ce1a4b18b;hb=61539aaf02d0537fd1df08b5d5bd03189992cf1e;hp=7888625e1c4d898d7f431e04db9cf2bdf6c9d938;hpb=fe637b465179cf2a217bbfc76d12bc41551b4ac4;p=sbp.git

diff --git a/src/edu/berkeley/sbp/Result.java b/src/edu/berkeley/sbp/Result.java
index 7888625..10a9b94 100644
--- a/src/edu/berkeley/sbp/Result.java
+++ b/src/edu/berkeley/sbp/Result.java
@@ -9,7 +9,7 @@ import java.io.*;
 import java.util.*;
 import java.lang.reflect.*;
 
-class Result implements GraphViz.ToGraphViz {
+final class Result implements GraphViz.ToGraphViz {
 
     private Forest f;
     private Node parent;
@@ -17,21 +17,23 @@ class Result implements GraphViz.ToGraphViz {
     private Position reduction;
     private HashSet<Node> children = new HashSet<Node>();
     private boolean destroyed = false;
+    private int usedByNonDoomedNode = 0;
 
     public Position reduction() { return reduction; }
     public GSS.Phase phase() { return phase; }
     public Forest getForest() { return f; }
     public Node parent() { return parent; }
-    public void addChild(Node child) { children.add(child); }
-    public void removeChild(Node child) { children.remove(child); }
+    public void addChild(Node child) {
+        children.add(child);
+        usedByNonDoomedNode += child.state().doomed ? 0 : 1;
+    }
+    public void removeChild(Node child) {
+        children.remove(child);
+        usedByNonDoomedNode -= child.state().doomed ? 0 : 1;
+    }
 
     public boolean usedByAnyNode() { return children.size() > 0; }
-    public boolean usedByNonDoomedNode() {
-        for(Node n : children)
-            if (!n.state().doomed)
-                return true;
-        return false;
-    }
+    public boolean usedByNonDoomedNode() { return usedByNonDoomedNode > 0; }
 
     public String toString() { return super.toString()+"->"+parent(); }