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
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
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)
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)
