/
MyBot.py
executable file
·151 lines (129 loc) · 4.46 KB
/
MyBot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
#!/usr/bin/env python
from __future__ import division
from sys import stdin
from copy import copy
"""
Strategies to implement:
* Create a list of high priority planets for me and for the enemy
* expansion phase when the enemy does not have that many planets (i.e., at beginning) or when the enemy does not have much strength
* Calculate the potential growth rate and cost of taking over any planet
* Calculate the risk of any of my planets, based on the fleets coming in. Determine if I should "save" the planet or evacuate the planet.
* Move headquarters if needed (maybe at every stage, any planet with more than X ships could move 3/4 of them to another planet to try to take over it?)
Done
* Using the knowledge about fleets, predict the game state in the future
* evacuate planet if it is doomed
* Calculate the growth rate for me for every planet (i.e., how many ships it is producing for me). Enemy planets count as negative growth rate. Neutral planets count as 0 growth rate.
Philosophies:
* Getting a planet to produce is more important than battling enemy ships.
* Prevent the enemy from becoming too strong
"""
from PlanetWars import PlanetWars, log, predict_state
def do_turn(pw):
log('planets: %s'%[(p.id, p.num_ships) for p in pw.my_planets])
log('fleets: %s'%[(f.num_ships, f.source, f.destination) for f in pw.my_fleets])
pw_future=predict_state(pw,MAX_TURNS)
for p in pw.planets:
p.future=[i.planets[p.id].num_ships if i.planets[p.id].owner==1 \
else -i.planets[p.id].num_ships for i in pw_future]
if pw.my_production >= 1.5*pw.enemy_production:
num_fleets=2
else:
num_fleets=4
if len(pw.my_fleets) >= num_fleets:
return
#log('finding source')
# (2) Find my strongest planet.
source = -1
source_score = -999999.0
source_num_ships = 0
s=None
dest=None
for p in pw.my_planets:
score = float(p.num_ships)/(1+p.growth_rate)
if score > source_score:
source_score = score
source = p.id
s=p
source_num_ships = p.num_ships
if s is not None:
#log('finding dest')
# (3) Find the weakest enemy or neutral planet.
dest = -1
dest_score = -999999.0
not_my_planets=set(pw.planets)-pw.my_planets
for p in not_my_planets:
score = float(1+p.growth_rate) / (1+p.num_ships)/(1+pw.distance(s,p))
if score > dest_score and not any(f.destination==p.id for f in pw.my_fleets):
dest_score = score
dest = p.id
#log('sending')
# (4) Send half the ships from my strongest planet to the weakest
# planet that I do not own.
num_ships=0
if source >= 0 and dest >= 0:
num_ships = source_num_ships / 2
pw.order(source, dest, num_ships)
my_planets=copy(pw.my_planets)
evacuate=set()
# if any of my planets is dying on the next turn, evacuate
for p in my_planets:
new_owner=pw_future[1].planets[p.id].owner
if new_owner!=1:
evacuate.add(p)
my_planets-=evacuate
if len(my_planets)==0:
my_planets=set([pw.planets[0]])
for p in evacuate:
log('evacuating %d to %d'%(p.id,dest))
if source==p.id:
p.num_ships-=num_ships
if p.num_ships>0:
dest=min(my_planets, key=lambda x: pw.distance(p,x))
pw.order(p.id, dest.id, p.num_ships)
MAX_TURNS=None
#from itertools import product
# needed because they only support python 2.5!
def product(*args, **kwds):
# product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
# product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
pools = map(tuple, args) * kwds.get('repeat', 1)
result = [[]]
for pool in pools:
result = [x+[y] for x in result for y in pool]
for prod in result:
yield tuple(prod)
def main():
global MAX_TURNS
map_data = ''
log('*'*30)
turn=0
while True:
current_line = stdin.readline()
if len(current_line) >= 2 and current_line.startswith("go"):
log('Turn %d '%turn+'='*30+'Turn %d'%turn)
pw = PlanetWars()
pw.parse_game_state(map_data)
if MAX_TURNS is None:
for p,q in product(pw.planets,repeat=2):
d=pw.distance(p,q)
if d>MAX_TURNS:
MAX_TURNS=d
if MAX_TURNS>10:
MAX_TURNS=10
log("predicting %s turns in the future"%MAX_TURNS)
do_turn(pw)
pw.finish()
map_data = ''
turn+=1
else:
map_data += current_line
if __name__ == '__main__':
try:
import psyco
psyco.full()
except ImportError:
pass
try:
main()
except KeyboardInterrupt:
print 'ctrl-c, leaving ...'