summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xstart.sh2
-rwxr-xr-xstrategy.py223
2 files changed, 225 insertions, 0 deletions
diff --git a/start.sh b/start.sh
new file mode 100755
index 0000000..5cd8bd7
--- /dev/null
+++ b/start.sh
@@ -0,0 +1,2 @@
+#! /bin/bash
+python strategy.py < /dev/stdin
diff --git a/strategy.py b/strategy.py
new file mode 100755
index 0000000..ebbf08d
--- /dev/null
+++ b/strategy.py
@@ -0,0 +1,223 @@
+#!/bin/python
+
+import sys, re
+
+entity=0
+memory = []
+memory_nest = []
+pheromone_type = 0
+see_food_id = 0
+see_food_zone = ""
+see_food_distance = 0
+see_food_amount = 0
+see_nest_id = 0
+see_nest_zone = ""
+see_nest_dist = 0
+see_nest_friend = 0
+ant_stock = 0
+ant_stamina = 0
+ant_sorted = 0
+ant_exist = 0
+explo = 0
+
+def log(string):
+ with open('out.txt', 'a') as f:
+ f.write(str(string)+"\n")
+
+for line in sys.stdin:
+ line = line.strip()
+
+ if line == "BEGIN ANT":
+ food_see_list = []
+ nest_see_list = []
+ pheromone_see_list = []
+ entity = 1
+ if line == "BEGIN NEST":
+ entity = 2
+ if line.startswith("MEMORY"):
+ line_splited = line.split()
+ m = []
+
+ m.append(line_splited[1])
+ m.append(line_splited[2])
+
+ if entity == 1:
+ if len(memory) == 0:
+ memory.append(m[0])
+ memory.append(m[1])
+ else:
+ memory[0] = m[0]
+ memory[1] = m[1]
+ elif entity == 2:
+ if len(memory_nest) == 0:
+ for x in range(0, 21):
+ if x != 0:
+ memory_nest.append(line_splited[x])
+ else:
+ for x in range(0, 21):
+ if x != 0:
+ memory_nest[x-1] = line_splited[x]
+ if line.startswith("SEE_FOOD"):
+ line_splited = line.split()
+ see_food_id = line_splited[1]
+ see_food_zone = line_splited[2]
+ see_food_distance = line_splited[3]
+ see_food_amount = line_splited[4]
+ liste = []
+ liste.append(see_food_id)
+ liste.append(see_food_zone)
+ liste.append(see_food_distance)
+ liste.append(see_food_amount)
+ food_see_list.append(liste)
+ if line.startswith("SEE_NEST"):
+ line_splited = line.split()
+ see_nest_id = line_splited[1]
+ see_nest_zone = line_splited[2]
+ see_nest_dist = line_splited[3]
+ see_nest_friend = line_splited[4]
+ liste = []
+ liste.append(see_nest_id)
+ liste.append(see_nest_zone)
+ liste.append(see_nest_dist)
+ liste.append(see_nest_friend)
+ nest_see_list.append(liste)
+ if line.startswith("SEE_PHEROMONE"):
+ line_splited = line.split()
+ see_pheromone_id = line_splited[1]
+ see_pheromone_zone = line_splited[2]
+ see_pheromone_dist = line_splited[3]
+ see_pheromone_type = line_splited[4]
+ liste = []
+ liste.append(see_pheromone_id)
+ liste.append(see_pheromone_zone)
+ liste.append(see_pheromone_dist)
+ liste.append(see_pheromone_type)
+ pheromone_see_list.append(liste)
+ if line.startswith("STOCK"):
+ line_splited = line.split()
+ ant_stock = line_splited[1]
+ if line.startswith("STAMINA"):
+ line_splited = line.split()
+ ant_stamina = line_splited[1]
+ if line == "END":
+ if entity == 1:
+ if int(ant_stock) < 10 and int(memory[1]) < 255:
+ if int(memory[0]) >= 8:
+ print("PUT_PHEROMONE "+str(memory[1]))
+ memory[0]=0
+ if int(memory[1]) < 3:
+ memory[1]=str(int(memory[1])+1)
+ else:
+ memory[1] = "1"
+ print("SET_MEMORY 0 "+str(memory[1]))
+ elif see_food_zone != "":
+ dist_close_food = 100
+ id_food = 0
+ probably_near_food_zone = ""
+ food_lvl = 0
+ for food_see in food_see_list:
+ if int(food_see[2]) <= int(dist_close_food):
+ dist_close_food = food_see[2]
+ id_food = food_see[0]
+ food_lvl = food_see[3]
+ probably_near_food_zone = food_see[1]
+ if probably_near_food_zone == "NEAR" and int(id_food) != 0 and int(food_lvl) > 10:
+ print("COLLECT "+id_food+" 10")
+ elif probably_near_food_zone == "FAR" and int(id_food) != 0:
+ print("MOVE_TO "+str(id_food))
+ elif see_food_zone == "":
+ print("EXPLORE")
+ memory[0]=str(int(memory[0])+1)
+ print("SET_MEMORY "+str(memory[0])+" "+str(memory[1]))
+ elif int(memory[1]) >= 255:
+ print("SUICIDE")
+ else:
+ if int(ant_stamina) <= 10:
+ print("EAT")
+ else:
+ nest_found_friend = 0
+ nest_friend_zone = "FAR"
+ nest_friend_id = 0
+ for nest_see in nest_see_list:
+ if nest_see[3] == "FRIEND":
+ nest_found_friend = 1
+ nest_friend_zone = nest_see[1]
+ nest_friend_id = nest_see[0]
+ if nest_found_friend == 0:
+ dist_closest_pheromone = 100
+ id_pheromone = 0
+ probably_near_food_zone = ""
+ no_pheromone = 0
+ type_pheromone = ""
+ dist_closest_pheromone_good_type = 100
+ id_pheromone_good_type = 0
+ probably_near_food_zone_good_type = ""
+ no_pheromone_good_type = 0
+ type_pheromone_good_type = 0
+
+ if int(memory[1]) > 1:
+ m1tmp=int(memory[1])-1
+ else:
+ m1tmp=3
+
+
+ log(pheromone_see_list)
+ pheromone_see_good_type_list = []
+ for pheromone_see in pheromone_see_list:
+ if str(pheromone_see[3]) == str(m1tmp):
+ pheromone_see_good_type_list.append(pheromone_see)
+ if pheromone_see_good_type_list != []:
+ near_phero = min(pheromone_see_good_type_list, key=lambda p: p[2])
+ type_pheromone_good_type = near_phero[3]
+ id_pheromone_good_type = near_phero[0]
+ dist_closest_pheromone_good_type = near_phero[2]
+ log(str(id_pheromone_good_type)+": "+str(dist_closest_pheromone_good_type))
+ if pheromone_see_list != []:
+ near_phero = min(pheromone_see_list, key=lambda p: p[2])
+ type_pheromone = near_phero[3]
+ id_pheromone = near_phero[0]
+ dist_closest_pheromone = near_phero[2]
+ log(str(id_pheromone)+": "+str(dist_closest_pheromone))
+
+ log(str(id_pheromone_good_type)+" != 0"+str(type_pheromone_good_type)+" == "+str(m1tmp))
+ log(str(id_pheromone)+" != 0"+str(id_pheromone_good_type)+" == 0")
+ if str(id_pheromone_good_type) != "0" and str(type_pheromone_good_type) == str(m1tmp):
+ log("good type:"+str(id_pheromone_good_type)+":"+str(dist_closest_pheromone_good_type)+" > 1")
+ if int(dist_closest_pheromone_good_type) > 1:
+ print("MOVE_TO "+id_pheromone_good_type)
+ no_pheromone = 0
+ else:
+ print("SET_MEMORY "+str(memory[0])+" "+str(m1tmp))
+ no_pheromone = 0
+ elif str(id_pheromone) != "0" and str(id_pheromone_good_type) == "0":
+ log(str(id_pheromone)+":"+str(dist_closest_pheromone)+" > 1")
+ if int(dist_closest_pheromone) > 1:
+ print("MOVE_TO "+id_pheromone)
+ no_pheromone = 0
+ else:
+ print("SET_MEMORY "+str(memory[0])+" "+str(m1tmp))
+ no_pheromone = 0
+ else:
+ no_pheromone = 1
+ if no_pheromone == 1:
+ if int(memory[0]) == 10:
+ print("EXPLORE")
+ explo = 1
+ else:
+ print("TURN 180")
+ print("SET_MEMORY 10 "+str(memory[1]))
+ no_pheromone == 0
+ else:
+ if nest_friend_zone == "NEAR":
+ print("NEST "+nest_friend_id)
+ print("SET MEMORY 0 "+str(memory[1]))
+ elif nest_friend_id != 0:
+ print("MOVE_TO "+nest_friend_id)
+ elif entity == 2:
+ if memory_nest[0] == "0" and memory_nest[1] == "0":
+ print("ANT_NEW 0")
+ print("SET_MEMORY 0 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0")
+ elif memory_nest[0] == "0" and memory_nest[1] == "1":
+ print("ANT_OUT 0 0 0 1")
+ print("SET_MEMORY 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0")
+ print("END", flush=True)