adventofcode2022/day7.py at main · lifeunleaded/adventofcode2022 · 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
#!/usr/bin/env python3
indatafile = open('day7_input')
indata = indatafile.readlines()
indatafile.close()
dirs = {'/':{'files':[],'dirs':[],'size':0,'parent':'/'}}
parents = {'/':''}
sizes = {'/':0}

thisdir = '/'
parentdir = '/'
for line in indata:
    if line.strip() == '$ cd ..':
        thisdir = parentdir
        parentdir = parents[thisdir]
    elif line[0:4] == '$ cd':
        parentdir = thisdir
        thisdir = parentdir+'/'+line[4:].strip()
        if thisdir not in dirs.keys():
            dirs[thisdir] = {'files':[],'dirs':[],'size':0}
            parents[thisdir] = parentdir
        if thisdir not in sizes.keys():
            sizes[thisdir] = 0
    elif line[0:3] == 'dir':
        parents[line[4:].strip()] = thisdir
        dirs[thisdir]['dirs'].append(line[4:].strip())
    elif line.split()[0].isnumeric():
        dirs[thisdir]['files'].append(line.split()[1].strip())
        dirs[thisdir]['size'] += int(line.split()[0])
        countupdir = thisdir
        while countupdir != '/':
            sizes[countupdir] += int(line.split()[0])
            countupdir = parents[countupdir]
        sizes['/']+= int(line.split()[0])
    else:
        pass

totalspace = 70000000
neededspace = 30000000
usedspace = sizes['/']
availablespace = totalspace - usedspace
missingspace = neededspace - availablespace
sizelist = [sizes[x] for x in sizes.keys()]
print(sum(filter(lambda size: size <= 100000, sizelist))) #part 1
print(min(filter(lambda size: size >= missingspace, sizelist))) #part 2