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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
/*
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
*/
#include <iostream>
#include <vector>
#include <cassert>
#include <cctype>
#define MIN(a,b) ((a)<(b)?(a):(b))
// triangle (stored as a square)
class Triangle {
public:
bool parse(int argc, char* argv[]) {
for (int row = 0; row < argc; row++) {
add_row();
if (!parse_row(row, argv[row]))
return false;
}
return true;
}
int min_sum() {
return min_sum_1(0, 0, 0);
}
private:
std::vector<std::vector<int>> m_data;
void add_row() {
int rows = static_cast<int>(m_data.size()) + 1;
m_data.resize(rows);
for (int i = 0; i < rows; i++)
m_data[i].resize(rows);
}
bool parse_row(int row, const char* text) {
int rows = static_cast<int>(m_data.size());
assert(row >= 0 && row < rows);
for (int i = 0; i < rows; i++) {
while (*text && !isdigit(*text)) text++;
if (!isdigit(*text))
return false;
m_data[row][i] = atoi(text);
while (*text && isdigit(*text)) text++;
}
return true;
}
int min_sum_1(int sum, int row, int col) {
sum += m_data[row][col];
int rows = static_cast<int>(m_data.size());
if (row + 1 == rows)
return sum;
else {
int sum1 = min_sum_1(sum, row + 1, col);
int sum2 = min_sum_1(sum, row + 1, col + 1);
return MIN(sum1, sum2);
}
}
};
int main(int argc, char* argv[]) {
if (argc == 1) {
fputs("Usage: ch-2 row row ...", stderr);
return EXIT_FAILURE;
}
argc--; argv++;
Triangle triangle;
if (!triangle.parse(argc, argv)) {
fputs("Malformed triangle", stderr);
return EXIT_FAILURE;
}
int sum = triangle.min_sum();
std::cout << sum << std::endl;
}
|