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
|
impl Solution {
pub fn add_operators(num: String, target: i32) -> Vec<String> {
fn dfs(res: &mut Vec<String>, num: &str, target: i64, expr: String, calc_val: i64, pre_num: i64, pos: usize) {
if pos == num.len() {
if calc_val == target {
res.push(expr);
}
return;
}
for i in pos..num.len() {
if i != pos && num.as_bytes()[pos] == b'0' {
break;
}
let cur_num = num[pos..=i].parse::<i64>().unwrap();
if pos == 0 {
dfs(res, num, target, cur_num.to_string(), cur_num, cur_num, i + 1);
} else {
dfs(res, num, target, format!("{}+{}", expr, cur_num), calc_val + cur_num, cur_num, i + 1);
dfs(res, num, target, format!("{}-{}", expr, cur_num), calc_val - cur_num, -cur_num, i + 1);
dfs(res, num, target, format!("{}*{}", expr, cur_num), calc_val - pre_num + pre_num * cur_num, pre_num * cur_num, i + 1);
}
}
}
let mut res = Vec::new();
dfs(&mut res, &num, target as i64, String::new(), 0, 0, 0);
res
}
}
|