add saboteur optimization problem. minor compile fixes.

saboteur.cpp

@@ -0,0 +1,158 @@
+#include <cstdio>
+#include <vector>
+#include <algorithm>
+
+#include "de.h"
+
+#define MAX(a,b)          ((a)>(b)?(a):(b))
+#define MIN(a,b)          ((a)<(b)?(a):(b))
+#define CLAMP(min,max,x)  (MAX(min,MIN(max,x)))
+
+class Saboteur: public DE::Engine<16>
+{
+	public:
+		double TestCardMix(int players, int saboteur_cards, int digger_cards)
+		{
+			int total_games = 0;
+			int total_saboteurs = 0;
+			int total_diggers = 0;
+			double saboteur_frequency = 0.0;
+
+			// make deck
+			std::vector<int> deck;
+			for (int i=0;i<saboteur_cards;++i)
+			{
+				deck.push_back(1);
+			}
+			for (int i=0;i<digger_cards;++i)
+			{
+				deck.push_back(0);
+			}
+
+			const int NUM_ITER = 5000;
+			for (int i=0;i<NUM_ITER;++i)
+			{
+				// shuffle deck
+				std::random_shuffle(deck.begin(),deck.end());
+
+				// deal first N cards
+				for (int j=0;j<players;++j)
+				{
+					if ( deck[j] == 1 )
+						++total_saboteurs;
+				}
+			}
+
+			return ((double)total_saboteurs/(double)NUM_ITER) / (double)players;
+		}
+
+		double TestIndividualFitness(int players, int saboteur_cards, int digger_cards, bool verbose=false)
+		{
+			double error_bonus = 0.0;
+			if ( saboteur_cards < 1 || saboteur_cards > 4 )
+				error_bonus += 1000.0;
+			if ( digger_cards < 1 || digger_cards > 7 )
+				error_bonus += 1000.0;
+			if ( saboteur_cards >= players )
+				error_bonus += 500.0;
+			if ( (saboteur_cards+digger_cards) < (players+1) )
+				error_bonus += 500.0;
+			if ( (saboteur_cards+digger_cards) > (players+3) )
+				error_bonus += 500.0;
+
+			if ( verbose )
+			{
+				printf("\n %2d players (%d, %d): ",players,saboteur_cards,digger_cards);
+			}
+
+			if ( error_bonus > 0.0 )
+			{
+				if ( verbose )
+				{
+					printf(" ERROR!  (%f)\n",error_bonus);
+				}
+				return error_bonus;
+			}
+			else
+			{
+				double actual = TestCardMix(players,saboteur_cards,digger_cards);
+				if ( verbose )
+				{
+					printf(" %0.5f\n",actual);
+				}
+				return actual;
+			}
+		}
+
+		double TestFitness(const double testcase[16], bool& stop)
+		{
+			ErrorAccumulator e;
+
+			const int num_players[8] = { 3, 4, 5, 6, 7, 8, 9, 10 };
+			const double expected = 0.363;
+			//const double expected = 1.0/3.0; // saboteurs are 1:3
+
+			for (int i=0;i<8;++i)
+			{
+				int i_saboteur_cards = i * 2;
+				int i_digger_cards   = i_saboteur_cards + 1;
+				int players          = num_players[i];
+
+				int saboteur_cards = (int)testcase[i_saboteur_cards];
+				int digger_cards   = (int)testcase[i_digger_cards];
+
+				double actual = TestIndividualFitness(players,saboteur_cards,digger_cards);
+
+				e.AddTestCase(expected,actual);
+			}
+
+			return e.GetRMSE();
+		}
+};
+
+int main(int argc, char** argv)
+{
+	// init
+	Saboteur s;
+	for (int i=0;i<8;++i)
+	{
+		s.SetRange(  i*2, 1.0, 4.0);
+		s.SetRange(i*2+1, 1.0, 7.0);
+	}
+	s.Reset();
+
+	// print original
+	const double original[16] = { 1,3,1,4,2,4,2,5,3,5,3,6,3,7,4,7 };
+	for (int i=0;i<8;++i)
+	{
+		const int num_players[8] = { 3, 4, 5, 6, 7, 8, 9, 10 };
+		int i_saboteur_cards = i * 2;
+		int i_digger_cards   = i_saboteur_cards + 1;
+		int players          = num_players[i];
+
+		int saboteur_cards = (int)original[i_saboteur_cards];
+		int digger_cards   = (int)original[i_digger_cards];
+
+		s.TestIndividualFitness(players,saboteur_cards,digger_cards,true);
+	}
+
+	// run
+	s.Solve(1000);
+
+	// output
+	const double* best = s.GetBest();
+	for (int i=0;i<8;++i)
+	{
+		const int num_players[8] = { 3, 4, 5, 6, 7, 8, 9, 10 };
+		int i_saboteur_cards = i * 2;
+		int i_digger_cards   = i_saboteur_cards + 1;
+		int players          = num_players[i];
+
+		int saboteur_cards = (int)best[i_saboteur_cards];
+		int digger_cards   = (int)best[i_digger_cards];
+
+		s.TestIndividualFitness(players,saboteur_cards,digger_cards,true);
+	}
+
+	return 0;
+}