-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathparseMathExpression.js
More file actions
54 lines (43 loc) · 1.64 KB
/
parseMathExpression.js
File metadata and controls
54 lines (43 loc) · 1.64 KB
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
// Parse Math Expression
// "123 + 456 + (123 - 456) + 789" -> ["123", "+", "456", "+", ["123", "-", "456"], "+", "789"]
const split = str => str.replace(/\s+/g, "").split(/([+-/*])/).filter(Boolean);
function getItem(arr, access) {
let reference = arr;
for (const item of access) {
reference = reference[item];
}
return reference;
}
function parseMathExpression(exp) {
const res = [];
const access = [];
const get = getItem.bind(null, res, access);
let str = "";
for (let n = 0; n < exp.length; n++) {
const item = exp[n];
if (item === ")" && !access.length) throw new Error(`Invalid expression at index ${n}`);
if (item === "(") {
const arr = get();
arr.push(...split(str));
arr.push([]);
access.push(arr.length - 1);
str = "";
} else if (item === ")") {
get().push(...split(str));
access.pop();
str = "";
} else {
str += item;
}
}
res.push(...split(str));
return res;
}
console.log(parseMathExpression("1")); // ["1"]
console.log(parseMathExpression("1 + 2")); // ["1", "+", "2"]
console.log(parseMathExpression("2 - 1")); // ["2", "-", "1"]
console.log(parseMathExpression("3 * 1")); // ["3", "*", "1"]
console.log(parseMathExpression("3 / 1")); // ["3", "/", "1"]
console.log(parseMathExpression("4 + (1 + 3)")); // ["4", "+", ["1", "+", "3"]]
console.log(parseMathExpression("123 + 456 + (123 - 456) + 789")); // ["123", "+", "456", "+", ["123", "-", "456"], "+", "789"]
console.log(parseMathExpression(")(")); // Error: Invalid expression at index 0