ARTICLE DETAIL

资讯详情

深耕郑州网站建设与运营推广的一线实战洞察。

元宝 LeetCode 6.Z字形变换 Rust实现

元宝    LeetCode 6.Z字形变换 Rust实现 LeetCode 6「Z 字形变换」Rust 实现给两种常见写法模拟行号反弹最直观、面试首选和数学周期直采无模拟、纯下标。题目回顾把字符串按从上到下、再斜向上的 Z 字形排成num_rows行然后按行从左到右拼起来。num_rows 1或num_rows s.len()时直接原样返回。写法一模拟行号反弹推荐维护num_rows个String桶用一个step在1 / -1之间反弹把字符塞进当前行。impl Solution { pub fn convert(s: String, num_rows: i32) - String { let n num_rows as usize; // 边界1 行或行数大于等于字符串长度 if n 1 || n s.len() { return s; } let mut rows: VecString vec![String::new(); n]; let mut cur 0usize; let mut step: isize 1; for c in s.chars() { rows[cur].push(c); // 到顶或到底就反向 if cur 0 { step 1; } else if cur n - 1 { step -1; } // isize 过渡避免下溢 cur (cur as isize step) as usize; } rows.concat() } }时间O(n)空间O(n)。step用isize中转是为了让cur - 1在cur 0时不会 panic虽然逻辑上不会真减到负数。写法二数学周期直采更“Rusty”一个完整 Z 字周期是cycle 2 * (n - 1)。第0行和最后一行下标每次cycle。中间第r行交替取j r和j cycle - r。impl Solution { pub fn convert(s: String, num_rows: i32) - String { let n num_rows as usize; if n 1 || n s.len() { return s; } let chars: Vecchar s.chars().collect(); let cycle 2 * (n - 1); let mut ans String::with_capacity(s.len()); for r in 0..n { let mut i r; let mut toggle true; // 中间行交替两种步长 while i chars.len() { ans.push(chars[i]); if r 0 || r n - 1 { i cycle; } else { i if toggle { 2 * (n - 1 - r) } else { 2 * r }; toggle !toggle; } } } ans } }不用模拟方向直接按行收集同样O(n)时间、O(n)空间。提交注意LeetCode Rust 模板是impl Solution { pub fn convert(s: String, num_rows: i32) - String { ... } }两种写法都能过第一种更好讲思路第二种显得对下标规律更熟。要不要我再补一个用(0..n).chain((1..n-1).rev()).cycle()的迭代器一行流版本
返回列表