#include "robber.h"
#include <iostream>
#include <iterator>
+#include <queue>
#include <cmath>
using namespace std;
using namespace __gnu_cxx;
+struct FindPlayer : SPGoal{
+ int _limit;
+ const hash_map<string, Player>& _players;
+ PlayerType _type;
+ FindPlayer(const hash_map<string, Player>& players, PlayerType type, int limit = 0) : _players(players), _type(type), _limit(limit){}
+ int operator()(const SPInfo* node) const{
+ if (_limit > 0 && node->cost >= _limit)
+ return -1;
+ for(hash_map<string, Player>::const_iterator player = _players.begin();
+ player != _players.end(); ++player){
+ if (player->second.type == _type && player->second.location == node->name){
+ return 1;
+ }
+ }
+ return 0;
+ }
+};
+
string Robber::turn(){
hash_map<string,double> streets;
Intersection& inter = _intersections[_location];
for (hash_map<string,StreetType>::const_iterator street = inter.connections.begin();
street != inter.connections.end(); ++street){
- double goodness = 10;
- int curx = _intersections[_location].x;
- int cury = _intersections[_location].y;
+ if (street->second == car){
+ cerr << "Discarding: " << street->first << " since car is needed" << endl;
+ continue;
+ }
+ double goodness = 0;
+ Intersection& conInter = _intersections[street->first];
- int newx = _intersections[street->first].x;
- int newy = _intersections[street->first].y;
- for (hash_map<string, Player>::const_iterator player = _players.begin();
- player != _players.end(); ++player){
- if (player->first == _name)
- continue;
+ list<string> closestFootCop = shortestPath(street->first, cop_foot, FindPlayer(_players, cop_foot, 6), true);
+ list<string> closestCarCop = shortestPath(street->first, cop_car, FindPlayer(_players, cop_car, 5), true);
- int px = _intersections[player->second.location].x;
- int py = _intersections[player->second.location].y;
+ unsigned int closestCop = 0;
+ bool copInCar = false;
- double dist1 = sqrt(pow((curx-px),2.0)+pow((cury-py),2.0));
- double dist2 = sqrt(pow((newx-px),2.0)+pow((newy-py),2.0));
- /*cerr << "Original distance: " << dist1 << endl;
- cerr << "New distance: " << dist2 << endl;
- */
+ if (closestCarCop.size() < closestFootCop.size() && closestCarCop.size() > 0){
+ closestCop = closestCarCop.size();
+ copInCar = true;
+ }else
+ closestCop = closestFootCop.size();
+ if (closestCop > 0 && closestCop < 3){
+ cerr << "Cop " << closestCop << " intersections away." << endl;
+ continue;
+ }
- if (dist2 < 50 && dist2 < dist1)
- goodness *= 0.95;
+ priority_queue<double> banks;
+ for(hash_map<string,int>::const_iterator bank = _banks.begin();
+ bank != _banks.end(); ++bank){
+ //cerr << "Handling bank at: " << bank->first << endl;
+ if (bank->second > 0){
+ list<string> l = shortestPath(street->first, _type, SimpleSPGoal(bank->first));
+ if (l.size() < 1)
+ continue;
+ //list<string>::iterator i = l.begin();
+ //++i;
+ banks.push(bank->second/(pow(l.size(),4.0)));
+ }
+ }
+ //sort(banks.begin(),banks.end(),greater<double>());
- if (player->second.type == cop_foot && dist2 <= 3)
- goodness /= 100;
- else if (player->second.type == cop_car && dist2 <= 2)
- goodness /= 100;
+ for (unsigned int i = 0; i < 2 && banks.size() > 0;++i){
+ goodness += banks.top();
+ banks.pop();
+ }
+ cerr << "Goodness before cop: " << goodness << endl;
+ if (closestCop > 2){
+ cerr << "Cop " << closestCop << " intersections away." << endl;
+ goodness *= 1 - 1/(copInCar ? closestCop : closestCop - 1);
+ }
+ cerr << "Goodness after cop: " << goodness << endl;
+ if (conInter.type == bank){
+ cerr << "FOUND A BANK" << endl;
+ if (closestCop > 0 && closestCop < 4)
+ continue;
+ else if (_banks[street->first] > 0){
+ cerr << "No cop close to bank" << endl;
+ return street->first;
+ }
}
cerr << "Street: " << street->first << " goodness: " << goodness << endl;
}
streets[_oldLocation] /= 10;
- for(hash_map<string,int>::const_iterator bank = _banks.begin();
- bank != _banks.end(); ++bank){
- //cerr << "Handling bank at: " << bank->first << endl;
- if (bank->second > 0){
- list<string> l = shortestPath(_location, bank->first,_type);
- if (l.size() < 1)
- continue;
- list<string>::iterator i = l.begin();
- streets[*++i] += bank->second/(pow(5.0,double(l.size())));
- }
- }
/*cerr << "Using route: ";
copy(v.begin(), v.end(), ostream_iterator<string>(cerr, " : "));
cerr << endl;
*/
- string destination;
+ string destination = _location;
double goodness = 0;
for (hash_map<string,double>::const_iterator dest = streets.begin();
dest != streets.end(); ++dest){