diff --git a/main.py b/main.py index 52987ad..968992f 100644 --- a/main.py +++ b/main.py @@ -5,8 +5,6 @@ import sys print("#hashcode2016") -import read - X = 0 # Nb rows Y = 0 # Nb columns D = 0 # Nb drones @@ -100,6 +98,8 @@ for d in range(D): Di.append(dict((p, 0) for p in range(P))) Dd.append(0) +Out = [] # Drones commands + # Debug assert(len(Dp) == len(Di) == len(Dd) == D) @@ -165,10 +165,14 @@ def deliver(d, c, p, q): def wait(d, w=1): assert(d in Da) Dd[d] += w + Da.remove(d) print("Drone", d, "waits", w, "turn" + ('s' if w >= 2 else ''), "→", Dd[d]) + Out.append(str(d) + ' W ' + str(w)) + # Control def newTurn(): + global t t += 1 print("--- Turn", t) for d in Da: @@ -176,16 +180,17 @@ def newTurn(): assert(len(Da) == 0) for d in range(D): if Dd[d] <= t: - Da.push(d) - print("Drones", ", ".join(Da), "(", len(Da), ")", "are avaible") + Da.append(d) + print("Drones", ", ".join([str(d) for d in Da]), "(", len(Da), ")", "are avaible") def end(): print("--- End!") -Out = '' # Drones commands # IA +newTurn() +newTurn() # Output f = open(sys.argv[1] + 'o', 'w') -f.write(len(Out)/10 + '\n' + Out) +f.write(str(len(Out)) + '\n' + '\n'.join(Out)) f.close()