-
Notifications
You must be signed in to change notification settings - Fork 2
/
maximum-value-after-insertion.rs
44 lines (39 loc) · 1 KB
/
maximum-value-after-insertion.rs
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
#![allow(dead_code, unused, unused_variables, non_snake_case)]
fn main() {}
struct Solution;
impl Solution {
pub fn max_value(n: String, x: i32) -> String {
let mut r = Vec::with_capacity(n.len() + 1);
let mut n = n.as_bytes();
let mut nega = false;
if n[0] == b'-' {
nega = true;
n = &n[1..];
r.push(b'-');
}
let x = x as u8 + b'0';
let mut done = false;
for &i in n {
if done {
r.push(i);
} else {
if nega {
if x < i {
r.push(x);
done = true;
}
} else {
if x > i {
r.push(x);
done = true;
}
}
r.push(i);
}
}
if !done {
r.push(x);
}
unsafe { String::from_utf8_unchecked(r) }
}
}