如何更新String中的char?

时间:2017-11-06 07:54:58

标签: string rust

我正在尝试更新char中的String,但似乎无法做到这一点。

fn main() {
    let mut s = "poyo".to_string();
    // s[1] = 'i'; or s.get_mut(1) = 'i'; can't do either
    println!("{}", s); // expecting "piyo"
}

我知道为什么会这种情况发生(String没有实现IndexMut<usize>),但我不知道如何解决这个问题......

playground link

1 个答案:

答案 0 :(得分:4)

答案取决于你正在处理的String种类;如果您正在使用ASCII (这意味着每个字符的大小只有一个字节,您可以直接操作基础Vec<u8>),您可以执行以下操作:

fn main() {
    let mut s = "poyo".to_string();
    let mut bytes = s.into_bytes();
    bytes[1] = 'i' as u8;

    unsafe { s = String::from_utf8_unchecked(bytes) }

    println!("{}", s);
}

或:

fn main() {
    let mut s = "poyo".to_string();

    unsafe {
        let bytes = s.as_bytes_mut();
        bytes[1] = 'i' as u8;
    }

    println!("{}", s);
}

但是,如果您(可能)使用多字节字符(这是String未实现IndexMut甚至Index)的全部原因,那么安全这样做的方法是使用Chars迭代器,遍历它并根据其元素创建一个新的String

fn main() {
    let s = "poyo".to_string();
    let iter = s.chars();
    let mut new = String::new();

    for (i, mut c) in iter.enumerate() {
        if i == 1 { c = 'i'; }
        new.push(c);
    }

    println!("{}", new);
}