about summary refs log tree commit diff
path: root/scratch/facebook/recursion-and-dynamic-programming/valid-parens.py
blob: 56f2c0b27456c7a46d9cd7a08244ad4e4d7a8e3e (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 valid_parens(n):
    if n == 0:
        return []
    if n == 1:
        return ["()"]

    result = set()
    for x in valid_parens(n - 1):
        result.add("({})".format(x))
        result.add("(){}".format(x))
        result.add("{}()".format(x))
    return result

def valid_parens_efficient(n):
    result = []
    curr = [''] * n**2
    do_valid_parens_efficient(result, curr, 0, n, n)
    return result

def do_valid_parens_efficient(result, curr, i, lhs, rhs):
    if lhs == 0 and rhs == 0:
        result.append(''.join(curr))
    else:
        if lhs > 0:
            curr[i] = '('
            do_valid_parens_efficient(result, curr, i + 1, lhs - 1, rhs)
        if rhs > lhs:
            curr[i] = ')'
            do_valid_parens_efficient(result, curr, i + 1, lhs, rhs - 1)

# Avoids recursion by using either a stack or a queue. I think this version is
# easier to understand.
def valid_parens_efficient_2(n):
    result = []
    xs = []
    xs.append(('', n, n))
    while xs:
        curr, lhs, rhs = xs.pop()
        print(curr)
        if lhs == 0 and rhs == 0:
            result.append(''.join(curr))
        if lhs > 0:
            xs.append((curr + '(', lhs - 1, rhs))
        if rhs > lhs:
            xs.append((curr + ')', lhs, rhs - 1))
    return result

# print(valid_parens(4))
print(valid_parens_efficient(3))
print(valid_parens_efficient_2(3))