aboutsummaryrefslogtreecommitdiff
path: root/challenge-100/paulo-custodio/python/ch-2.py
blob: 0d22eba8d13f44924f23139c0e7572922dd8c617 (plain)
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
#!/usr/bin/env python

# Challenge 100
#
# TASK #2 > Triangle Sum
# Submitted by: Mohammad S Anwar
# You are given triangle array.
#
# Write a script to find the minimum path sum from top to bottom.
#
# When you are on index i on the current row then you may move to either
# index i or index i + 1 on the next row.
#
# Example 1:
# Input: Triangle = [ [1], [2,4], [6,4,9], [5,1,7,2] ]
# Output: 8
#
# Explanation: The given triangle
#
#             1
#            2 4
#           6 4 9
#          5 1 7 2
#
# The minimum path sum from top to bottom:  1 + 2 + 4 + 1 = 8
#
#              [1]
#            [2]  4
#            6 [4] 9
#           5 [1] 7 2
# Example 2:
# Input: Triangle = [ [3], [3,1], [5,2,3], [4,3,1,3] ]
# Output: 7
#
# Explanation: The given triangle
#
#             3
#            3 1
#           5 2 3
#          4 3 1 3
#
# The minimum path sum from top to bottom: 3 + 1 + 2 + 1 = 7
#
#              [3]
#             3  [1]
#            5 [2] 3
#           4 3 [1] 3

import sys;

triangle = []

def add_row(row, items):
    triangle.append(items)

def parse(args):
    for i in range(0, len(args)):
        items = [int(x) for x in args[i].split(",")]
        add_row(i, items)

def min_sum():
    def min_sum_1(sum, row, col):
        sum += triangle[row][col]
        if row+1 == len(triangle):
            return sum
        else:
            sum1 = min_sum_1(sum, row+1, col)
            sum2 = min_sum_1(sum, row+1, col+1)
            return min(sum1, sum2)
    return min_sum_1(0, 0, 0)

parse(sys.argv[1:])
print(min_sum())