diff --git a/code_study/Baekjoon/python/16946.py b/code_study/Baekjoon/python/16946.py index 759e37e..45ea841 100644 --- a/code_study/Baekjoon/python/16946.py +++ b/code_study/Baekjoon/python/16946.py @@ -48,20 +48,10 @@ for x, y in wall : if 0 <= nx < M and 0 <= ny < N and area_num[ny][nx] != -1 : area = area_num[ny][nx] - - if len(neighbor) == 0 : - cnt += area_size[area] - neighbor.append(area) - else : - dupli_check = True + neighbor.append(area) - for n in neighbor : - if n == area : - dupli_check = False - - if dupli_check : - cnt += area_size[area] - neighbor.append(area) + for n in set(neighbor) : + cnt += area_size[n] res[y][x] = cnt % 10