Assignment1_advent/Day_06.py at main · wkai10998/Assignment1_advent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
#part1
#insight：因为有toggle所以不是简单加减法，需要有位置属性以及对应状态
#字典写法
#创建网格
grid=dict()
for x in range(1000):
    for y in range(1000):
        grid[(x, y)]=False
op=0
x1,x2,y1,y2=0,0,0,0
#清洗数据
with open('input/day_06.txt', 'r') as file_object:
    input_text = file_object.read()
    input_text = input_text.split('\n')
    for chunk in input_text:
        chunk = chunk.strip()
        if chunk:
            chunk_part = chunk.split(' ')
            if chunk_part[0]=="toggle":
                op = 'toggle'
                pos1 = chunk_part[1]
                pos2 = chunk_part[3]
            elif chunk_part[1]=="on":
                op = 'on'
                pos1 = chunk_part[2]
                pos2 = chunk_part[4]
            elif chunk_part[1]=="off":
                op = 'off'
                pos1 = chunk_part[2]
                pos2 = chunk_part[4]
    #定义规则
            pos_start = pos1.split(",")
            pos_end = pos2.split(",")
            x1 = int(pos_start[0])
            x2 = int(pos_end[0])
            y1 = int(pos_start[1])
            y2 = int(pos_end[1])
            for x in range(x1, x2 + 1):
              for y in range(y1, y2 + 1):
                if op == 'toggle':
                   grid[(x, y)] = not grid[(x, y)]
                elif op == 'on':
                   grid[(x, y)] = True
                elif op == 'off':
                   grid[(x, y)] = False
#计算数量
light_count = 0
for value in grid.values():
    if value:
        light_count += 1
print(light_count)


# #二维表格写法(wiht the help of AI)
# #第一步 创建原始坐标
# light_grid = []
# for x in range(1000):
#     one_row = []
#     for y in range(1000):
#         one_row.append(False)
#     light_grid.append(one_row)
# #第二步 清理数据
# with open('input/day_06.txt', 'r') as file_object:
#     clean_input=list()
#     for line in file_object:
#          clean_line = line.strip()
#          if clean_line: #防止有空行返回空值
#               clean_input.append(clean_line)
# #第三步 拆分数据，提取「操作类型」和「坐标字符串」
# for cmd in clean_input:
#     cmd_part = cmd.split(' ')
#     x1 = 0
#     y1 = 0
#     x2 = 0
#     y2 = 0
#     if cmd_part[0] == 'toggle':
#         op = 'toggle'
#         pos1 = cmd_part[1]
#         pos2 = cmd_part[3]
#     elif cmd_part[1] == 'on':
#         op = 'on'
#         pos1 = cmd_part[2]
#         pos2 = cmd_part[4]
#     elif cmd_part[1] == 'off':
#         op = 'off'
#         pos1 = cmd_part[2]
#         pos2 = cmd_part[4]
#     #把坐标字符串转为包含两个元素的列表
#     pos1_list = pos1.split(',')  # "0,0" → ["0", "0"]
#     x1 = int(pos1_list[0])
#     y1 = int(pos1_list[1])
#     pos2_list = pos2.split(',')
#     x2 = int(pos2_list[0])
#     y2 = int(pos2_list[1])
# # 第四步 计算坐标范围
#     # 遍历所有x：range左闭右开）
#     for current_x in range(x1, x2 + 1):
#         for current_y in range(y1, y2 + 1):
#             # 【5. 定义开关规则】
#             if op == 'on':
#                 light_grid[current_x][current_y] = True
#             elif op == 'off':
#                 light_grid[current_x][current_y] = False
#             elif op == 'toggle':
#                 if light_grid[current_x][current_y] == True:
#                     light_grid[current_x][current_y] = False
#                 else:
#                     light_grid[current_x][current_y] = True
#
# #第五步 统计最终亮着的灯数
#     on_count = 0
# for x in range(1000):
#     for y in range(1000):
#         if light_grid[x][y] == True:
#             on_count += 1
#
# print('最终亮起的灯总数：', on_count)

#part2
#创建网格
grid=dict()
for x in range(1000):
    for y in range(1000):
        grid[(x, y)]=0
op=0
x1,x2,y1,y2=0,0,0,0
#清洗数据
with open('input/day_06.txt', 'r') as file_object:
    input_text = file_object.read()
    input_text = input_text.split('\n')
    for chunk in input_text:
        chunk = chunk.strip()
        if chunk:
            chunk_part = chunk.split(' ')
            if chunk_part[0]=="toggle":
                op = 'toggle'
                pos1 = chunk_part[1]
                pos2 = chunk_part[3]
            elif chunk_part[1]=="on":
                op = 'on'
                pos1 = chunk_part[2]
                pos2 = chunk_part[4]
            elif chunk_part[1]=="off":
                op = 'off'
                pos1 = chunk_part[2]
                pos2 = chunk_part[4]
    #定义规则
            pos_start = pos1.split(",")
            pos_end = pos2.split(",")
            x1 = int(pos_start[0])
            x2 = int(pos_end[0])
            y1 = int(pos_start[1])
            y2 = int(pos_end[1])
            for x in range(x1, x2 + 1):
              for y in range(y1, y2 + 1):
                if op == 'toggle':
                   grid[(x, y)] += 2
                elif op == 'on':
                   grid[(x, y)] += 1
                elif op == 'off':
                   grid[(x, y)] = max(0, grid[(x, y)] - 1)
#计算数量
light_count = 0
for value in grid.values():
    if value > 0:
        light_count += value
print(light_count)