day21/part1.cpp


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

#include <cctype>
#include <iostream>
#include <map>
#include <string>
#include <vector>

struct Node
{
    std::string left;
    std::string right;
    char op = 0;
    long long n = 0;

    bool solved() const { return left.empty(); }

    bool solve(const auto& tree) {
        const auto& ln = tree.at(left);
        const auto& rn = tree.at(right);

        if (ln.solved() && rn.solved()) {
            switch (op) {
            case '+':
                n = ln.n + rn.n;
                break;
            case '-':
                n = ln.n - rn.n;
                break;
            case '*':
                n = ln.n * rn.n;
                break;
            case '/':
                n = ln.n / rn.n;
                break;
            }

            left.clear();
        }

        return solved();
    }
};

int main()
{
    std::map<std::string, Node> tree;

    while (!std::cin.eof()) {
        std::string line;
        std::getline(std::cin, line);
        if (std::cin.eof())
            break;

        auto split = line.find(':');
        const auto key = line.substr(0, split);
        line = line.substr(split + 2);

        Node node;
        if (isdigit(line.front())) {
            node.n = std::stoll(line);
        } else {
            split = line.find(' ');
            node.left = line.substr(0, split);
            node.op = line.at(split + 1);
            node.right = line.substr(split + 3, 4);
        }

        tree.emplace(key, node);
    }

    bool more = true;
    while (more) {
        more = false;
        for (auto& [k, v] : tree) {
            if (!v.solved())
                more |= !v.solve(tree);
        }
    }

    std::cout << "root: " << tree["root"].n << std::endl;

    return 0;
}