codemyself
unknown
python
3 years ago
7.2 kB
4
Indexable
import sys
import math
DEBUG = True
def debug(*args):
if (DEBUG):
print(args,
file=sys.stderr)
wiz1 = 0
MAP = [16000, 7500]
WIZARD_RADIUS = 400
SNAFFLE_RADIUS = 150
MY_TEAM_ID = int(input())
if MY_TEAM_ID == 0:
GOALS = [16000, 3750]
else:
GOALS = [0, 3750]
class Entities:
entities_id = None
entity_type = None
x = None
y = None
vx = None
vy = None
state = None
def __init__(self, data):
self.entities_id = data[0]
self.entity_type = data[1]
self.x = data[2]
self.y = data[3]
self.vx = data[4]
self.vy = data[5]
self.state = data[6]
class Player:
x = None
y = None
vx = None
vy = None
state = None
def __init__(self, data):
self.entities_id = data.entities_id
self.entity_type = data.entity_type
self.x = data.x
self.y = data.y
self.vx = data.vx
self.vy = data.vy
self.state = data.state
def get_distance_to(self, target):
return math.hypot(abs(self.x - target.x), abs(self.y - target.y))
def get_snaffle(self, all_entities):
min_dist_to_snaffle = float('inf')
snaffle = None
for target in all_entities:
if target.entity_type != 'SNAFFLE':
continue
if self.get_distance_to(target) <= min_dist_to_snaffle:
min_dist_to_snaffle = self.get_distance_to(target)
snaffle = target
return snaffle
def get_snaffle_id(self, all_entities):
min_dist_to_snaffle = float('inf')
snaffle = None
for target in all_entities:
if target.entity_type != 'SNAFFLE':
continue
if self.get_distance_to(target) <= min_dist_to_snaffle:
min_dist_to_snaffle = self.get_distance_to(target)
snaffle = target
return snaffle.entities_id
@staticmethod
def move(x, y, thruster):
print('MOVE %s %s %s' % (x, y, thruster))
@staticmethod
def throw(x, y, power):
print('THROW %s %s %s' % (x, y, power))
while True:
game_entities = []
myScore, myMagic = input().split()
opponentScore, opponentMagic = input().split()
entities = int(input())
for _ in range(entities):
entity_id, entity_type, x, y, vx, vy, state = input().split()
entitie = Entities([int(entity_id), entity_type, int(x), int(y), int(vx), int(vy), int(state)])
game_entities.append(entitie)
player_num = 1
my_wizard_1, my_wizard_2 = None, None
for entitie in game_entities:
if entitie.entity_type == 'WIZARD':
if player_num == 1:
player_num += 1
my_wizard_1 = Player(entitie)
elif player_num == 2:
my_wizard_2 = Player(entitie)
target_1 = my_wizard_1.get_snaffle(all_entities=game_entities)
target_2 = my_wizard_2.get_snaffle(all_entities=game_entities)
#if my_wizard_1.get_distance_to(target_1) > WIZARD_RADIUS:
#my_wizard_1.move(target_1.x, target_1.y, thruster=150)
#elif int(my_wizard_1.state) == 1:
#my_wizard_1.throw(GOALS[0], GOALS[1], power=500)
if my_wizard_1.get_distance_to(target_1) > WIZARD_RADIUS:
if int(myMagic) > 30:
if my_wizard_1.x >= 8000 and my_wizard_1.x <= 8000:
print('FLIPENDO %s' % my_wizard_1.get_snaffle_id(all_entities=game_entities))
elif MY_TEAM_ID == 0:
if my_wizard_1.x < 3000:
print('ACCIO %s' % my_wizard_1.get_snaffle_id(all_entities=game_entities))
else:
print('FLIPENDO %s' % my_wizard_1.get_snaffle_id(all_entities=game_entities))
else:
if my_wizard_1.x > 13000:
print('ACCIO %s' % my_wizard_1.get_snaffle_id(all_entities=game_entities))
else:
print('FLIPENDO %s' % my_wizard_1.get_snaffle_id(all_entities=game_entities))
#print('FLIPENDO %s' % my_wizard_1.get_snaffle_id(all_entities=game_entities))
else:
my_wizard_1.move(target_1.x, target_1.y, thruster=150)
elif int(my_wizard_1.state) == 1:
#if int(myMagic) > 20:
#print('FLIPENDO %s' % my_wizard_1.get_snaffle_id(all_entities=game_entities))
#else:
my_wizard_1.throw(GOALS[0], GOALS[1], power=500)
else:
my_wizard_1.move(target_1.x, target_1.y, thruster=150)
if my_wizard_2.get_distance_to(target_2) > WIZARD_RADIUS:
#if int(myMagic) > 20:
#if my_wizard_2.x >= 8000 and my_wizard_2.x <= 8000:
#print('FLIPENDO %s' % my_wizard_2.get_snaffle_id(all_entities=game_entities))
#elif MY_TEAM_ID == 0:
#if my_wizard_2.x < 5000:
#print('ACCIO %s' % my_wizard_2.get_snaffle_id(all_entities=game_entities))
#else:
#print('FLIPENDO %s' % my_wizard_2.get_snaffle_id(all_entities=game_entities))
#else:
#if my_wizard_2.x > 11000:
#print('ACCIO %s' % my_wizard_2.get_snaffle_id(all_entities=game_entities))
#else:
#print('FLIPENDO %s' % my_wizard_2.get_snaffle_id(all_entities=game_entities))
#else:
if MY_TEAM_ID == 0:
if my_wizard_2.get_distance_to(target_2) < 4000 and my_wizard_2.x < 4000 and target_2.x < 4000:
my_wizard_2.move(target_2.x, target_2.y, thruster=150)
if int(my_wizard_2.state) == 1:
my_wizard_2.throw(GOALS[0], GOALS[1], power=500)
elif int(myMagic) > 15:
if target_2.x < my_wizard_2.x:
if target_2.vx < 1200:
if target_2.x > 0 and target_2.x < 16000:
print('ACCIO %s' % my_wizard_2.get_snaffle_id(all_entities=game_entities))
else:
my_wizard_2.move(100, 3750, thruster=150)
elif MY_TEAM_ID == 1:
if my_wizard_2.get_distance_to(target_2) < 4000 and my_wizard_2.x > 12000 and target_2.x > 12000:
my_wizard_2.move(target_2.x, target_2.y, thruster=150)
if int(my_wizard_2.state) == 1:
my_wizard_2.throw(GOALS[0], GOALS[1], power=500)
#elif int(myMagic) > 20:
#if target_2.x > 0 and target_2.x < 16000:
#print('FLIPENDO %s' % my_wizard_2.get_snaffle_id(all_entities=game_entities))
else:
my_wizard_2.move(1500, 3750, thruster=150)
#my_wizard_2.move(target_2.x, target_2.y, thruster=150)
elif int(my_wizard_2.state) == 1:
my_wizard_2.throw(GOALS[0], GOALS[1], power=500)
else:
my_wizard_2.move(target_2.x, target_2.y, thruster=150)Editor is loading...