Rust编程小知识|Rust如何修改字符串String中的字符 rust|元组|rust|i32|rust

文章目录

- - - 借助remove()和insert()
    - 更高效的方式
    - 更进一步，自定义SetByIndex trait

节选自： 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 【Rust编程小知识|Rust如何修改字符串String中的字符】为了调用更加方便，我们完全可以自定义一个根据位置修改字符的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