algoright returns solution

Author: mackoo13

algorithm/algorithm.py

@@ -60,5 +60,5 @@ def execute(data):
  # podsumowujac - jesli np. n=5, m=3, e=1, to zawsze na koniec iteracji 1 pszczola pochodzi z obszaru
  # elitarnego, 2 z dobrych, a 2 sa losowane od nowa.
 
- print(max(pool, key=lambda s: calculate_score(data, s)))
- return calculate_score(data, pool[0])
+ pool = sorted(pool, key=lambda s: calculate_score(data, s), reverse=True)
+ return pool[0]

main.py

@@ -1,27 +1,32 @@
 from algorithm.algorithm import Algorithm
 from parsers.input_parser import parse_input
 from utils.solution_checker import check_solution
+from parsers.output_builder import save_solution
 import time
 
+
 def main():
  # input_file = "files/videos_worth_spreading.in"
  # input_file = "files/kittens.in"
  # input_file = "files/example1.in"
- input_file = "files/me_at_the_zoo.in"
- # input_file = "files/trending_today.in"
+ # input_file = "files/me_at_the_zoo.in"
+ input_file = "files/trending_today.in"
 
  # solution_file = "files/example1.out"
  # solution = check_solution(input_file, solution_file)
  # print(solution)
 
  before = time.time()
- score = Algorithm.execute(parse_input(input_file))
+ best_solution = Algorithm.execute(parse_input(input_file))
+ save_solution(best_solution, "files/out1.out")
+ score = check_solution(input_file, "files/out1.out")
  after = time.time()
 
+ print(best_solution)
  print("-----------------------------")
  print("Algorithm took: " + str(round(after-before, 2)) + " s")
  print("-----------------------------")
- print("Saved time: " + str(round(score / 1000, 2)) + " s")
+ print("Saved time: " + str(round(score / 1000, 2)) + " s (score: " + str(score) + ")")
 
 if __name__ == '__main__':
  main()

parsers/output_builder.py

@@ -0,0 +1,5 @@
+def save_solution(solution, name):
+ with open(name, 'w') as f:
+ f.write(str(len(solution)) + "\n")
+ for s in solution:
+ f.write(str(s) + " " + " ".join([str(v) for v in solution[s]]) + "\n")