일단 돌아는 가게 수정...

Author: HwangSB

ai_team_TNK.py

@@ -85,7 +85,7 @@ def generate_json(prefix, my_position, your_position):
 
         # json파일로 저장
         stone["index"] = my_idx
-        stone["positions"] = {'my':[], 'your':[]}
+        stone["positions"] = {'my': [], 'your': []}
         stone["strength"] = []
 
         #현재 위치 json에 추가
@@ -208,6 +208,7 @@ def get_high_score_cases(filenames, count):
         with open(str(filename)) as json_file:
             json_data = json.load(json_file)
 
+        priority = 0
         for i in range(len(json_data['result'])):
             # 내 남은 돌의 개수 계산
             my_stones = len(json_data['result'][i]['my'])
@@ -224,24 +225,26 @@ def get_high_score_cases(filenames, count):
             # stone_info 에 튜플 형태로 힙 push 
             heapq.heappush(stone_info, (
                     (-1) * int(score), # 점수
+                    priority,
                     json_data['index'], # 돌 번호
                     json_data['strength'][i]['x'], # 돌의 x 세기
                     json_data['strength'][i]['y'], # 돌의 y 세기
                     json_data['result'][i]['my'], # 내 돌의 포지션
                     json_data['result'][i]['your'], # 상대 돌의 포지션
                 )
             )
+            priority += 1
 
     cases = []
 
     for i in range(0,count):
         tmp = heapq.heappop(stone_info)
 
-        my_positions = list(map(lambda position: [position['x'], position['y']], tmp[4]))
-        your_positions = list(map(lambda position: [position['x'], position['y']], tmp[5]))
+        my_positions = list(map(lambda position: [position['x'], position['y']], tmp[5]))
+        your_positions = list(map(lambda position: [position['x'], position['y']], tmp[6]))
         positions = [my_positions, your_positions]
 
-        case = Case(tmp[0] * (-1), [tmp[2], tmp[3]], positions, tmp[1])
+        case = Case(tmp[0] * (-1), [tmp[3], tmp[4]], positions, tmp[2])
         cases.append(case)
 
     return cases

alggago.rb

@@ -9,7 +9,7 @@
 
 WIDTH, HEIGHT = 1000, 700
 TICK = 1.0/60.0
-NUM_STONES = 7
+NUM_STONES = 3
 PLAYER_COLOR = ["black", "white"]
 STONE_DIAMETER = 50
 RESTITUTION = 0.9