Changeset 12832 for branches/HeuristicLab.Problems.GrammaticalOptimization/HeuristicLab.Algorithms.MonteCarloTreeSearch/Base/TreeNode.cs

Timestamp:

08/03/15 00:03:18 (9 years ago)

Author:

aballeit

Message:

#2283 limit parallelism

File:

• branches/HeuristicLab.Problems.GrammaticalOptimization/HeuristicLab.Algorithms.MonteCarloTreeSearch/Base/TreeNode.cs (modified) (2 diffs)

branches/HeuristicLab.Problems.GrammaticalOptimization/HeuristicLab.Algorithms.MonteCarloTreeSearch/Base/TreeNode.cs

@@ -14 +14 @@
         public string phrase;
         public TreeNode parent;
-        public List<TreeNode> children;
+        public TreeNode[] children;
         public IBanditPolicyActionInfo actionInfo;
-        public int level;
+        public ushort level;
 
-        public TreeNode(TreeNode parent, string phrase, IBanditPolicyActionInfo actionInfo, int level)
+        public TreeNode(TreeNode parent, string phrase, IBanditPolicyActionInfo actionInfo, ushort level)
         {
             this.parent = parent;
@@ -34 +34 @@
             return children.Select(n => n.actionInfo);
         }
+
+        internal void RemoveChildren(TreeNode currentNode)
+        {
+            TreeNode[] newChildren = new TreeNode[children.Length-1];
+            int counter = 0;
+            for (int i = 0; i < children.Length; i++)
+            {
+                if (children[i] != currentNode)
+                {
+                    newChildren[counter] = children[i];
+                    counter++;
+                }
+            }
+            children = newChildren;
+        }
     }
 }