Rust如何修改字符串String中的字符

节选自： Rust 09: 字符串详解（String、&str、内存布局）

---

借助remove()和insert()

可以借助remove()和insert()，但是这2个方法都会导致字符串整体迁移，是O(n)复杂度的方法。如果字符串比较长，这样做代价很大。

// remove和insert都是O(n)复杂度的
// remove会导致删除位置后面的所有字符整体前移1个位置
// insert会导致插入位置后面的所有字符整体后移1个位置
let mut s1:String = String::from("Hello");
s1.remove(0);
s1.insert(0, 'X');
assert_eq!("Xello", s1);

更高效的方式

对于追求极致效率的rust而言，提供了unsafe关键字。这里有一个O(1)复杂度的unsafe方法：

// 如何高效修改String中的一个字符
let mut s1:String = String::from("Hello");
unsafe {
    let s1_bytes: &mut [u8] = s1.as_bytes_mut();
    s1_bytes[0] = 'X' as u8;
    println!("s1 new={}", s1);//s1 new=Xello
}

更进一步，自定义SetByIndex trait

为了调用更加方便，我们完全可以自定义一个根据位置修改字符的trait，并且为String类型实现这个trait。

trait SetByIndex {
    fn set_by_index(&mut self, idx: usize, c: u8);
}
impl SetByIndex for String {
    fn set_by_index(&mut self, idx: usize, c: u8) {
        if idx>=self.len() {
            panic!("Index out of bounds: {}, expected: [0,{})", idx, self.len());
        }
        unsafe {
            let _buf: &mut [u8] = self.as_bytes_mut();
            _buf[idx] = c;
        }
    }
}
let mut ss = "Hello".to_owned();
ss.set_by_index(8, 'X' as u8);
println!("ss={}", ss);//ss=Xello