Made Algorithm functions more generic.

[genetic.git] / src / main.cpp

diff --git a/src/main.cpp b/src/main.cpp
index 3816b435095ef27255cb76f50ab4192231da97c0..f2aec1380acfd130482c57cc364305fdb6da4876 100644 (file)
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -1,5 +1,7 @@
 #include <iostream>
 #include <vector>
+#include <map>
+#include <string>
 
 #include "gene.h"
 #include "chromosome.h"
@@ -27,7 +29,7 @@ int main() {
     typedef Mutation<_Chromosome> _Mutation;
 
     typedef generator::Generation<_Chromosome> _Generator;
-    typedef Algorithm<_Chromosome, _Selection, _Crossover, _Mutation> _Algorithm;
+    typedef Algorithm<_Chromosome, _Selection, _Crossover, _Mutation, _Fitness> _Algorithm;
     typedef Condition<_Chromosome> _Condition;
 
     const int chromosomeSize = 10;