about summary refs log tree commit diff
path: root/scratch/facebook/language2.py
blob: 3aebd454833b7b0f5b45a21146d3b80b001f11cd (plain) (blame)
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









def tokenize(xs):
    result = []
    i = 0
    while i < len(xs):
        curr = xs[i]
        if curr in {'*','+'}:
            result.append(curr)
            i += 1
            continue
        i += 1
        while i < len(xs) and xs[i] in {str(x) for x in range(10)}:
            curr += xs[i]
            i += 1
        result.append(int(curr))
    return result

def parse(tokens):
    result = []
    series = []
    for token in tokens:
        if token == '*':
            continue
        elif token == '+':
            result.append(series)
            series = []
        else:
            series.append(token)
    if series:
        result.append(series)
    return result

def product(xs):
    result = 1
    for x in xs:
        result *= x
    return result

def evaluate(tree):
    return sum([product(xs) for xs in tree])

print(evaluate(parse(tokenize("2+30*8*9+10"))))