planar_graph_playground/python/util.py at main · Hermann-SW/planar_graph_playground · 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
import math
import numpy as np
assert math and np
#ifndef __util__
#define __util__

def filled_array(n, m, v=0):
    A = []
    for _ in range(n):
        if m == 1:
            A.append(v)
        else:
            a = []
            for _ in range(m):
                a.append(v)
            A.append(a)
    return A

def incr(arr, i=0):
    arr[i] += 1

def aset(arr, x, i=0):
    arr[i] = x

def parse(st):
    result = []
    i = 0

    while st[i] != '[':
        i += 1

    while st[i] != ']':

        i += 1
        while st[i].isspace():
            i += 1

        j = i

        while not(st[i].isspace()) and (st[i] != ']') and (st[i] != ','):
            i += 1

        result.append(int(st[j:i]))

        while st[i] != ']' and st[i] != ',':
            i += 1

    return st[i+1:],result

def parse2(st):
    result = []
    i = 0

    while st[i] != '[':
        i += 1

    while st[i] != ']':

        i += 1
        while st[i].isspace():
            i += 1

        st, p = parse(st[i:])
        result.append(p)

        i = 0
        while st[i] != ']' and st[i] != ',':
            i += 1

    return st[i+1:],result

def parse3(st):
    result = []
    i = 0

    while st[i] != '[':
        i += 1

    while st[i] != ']':

        i += 1
        while st[i].isspace():
            i += 1

        st, p = parse2(st[i:])
        result.append(p)

        i = 0
        while st[i] != ']' and st[i] != ',':
            i += 1

    return result

def parse2file(name):
    return eval(open(name).read())

#endif