Iterator::collect 是否分配与 String::with_capacity 相同的内存量？

Question

在 C++ 中，当连接一堆字符串（其中每个元素的大小大致已知）时，通常会预先分配内存以避免多次重新分配和移动：

std::vector<std::string> words;
constexpr size_t APPROX_SIZE = 20;

std::string phrase;
phrase.reserve((words.size() + 5) * APPROX_SIZE);  // <-- avoid multiple allocations
for (const auto &w : words)
  phrase.append(w);

同样，我在 Rust 中做了这个（这个块需要 unicode-segmentation crate）

fn reverse(input: &str) -> String {
    let mut result = String::with_capacity(input.len());
    for gc in input.graphemes(true /*extended*/).rev() {
        result.push_str(gc)
    }
    result
}

有人告诉我，惯用的表达方式是单一的表达方式

fn reverse(input: &str) -> String {
  input
      .graphemes(true /*extended*/)
      .rev()
      .collect::<Vec<&str>>()
      .concat()
}

虽然我真的很喜欢它并想使用它，但从内存分配的角度来看，前者分配的块会比后者少吗？

我用cargo rustc --release -- --emit asm -C "llvm-args=-x86-asm-syntax=intel"反汇编了它，但它没有穿插源代码，所以我很茫然。

Answer 1

您的原始代码很好，我不建议更改它。

原版分配一次：inside String::with_capacity。

第二个版本分配至少两次：首先，它创建一个Vec<&str> 并通过pushing &strs 来增长它。然后，它计算所有&strs 的总大小并创建一个具有正确大小的新String。 （此代码在the join_generic_copy method in str.rs 中。）这很糟糕有几个原因：

1. 显然，它不必要地分配。
2. 字素簇可以任意大，因此中间的Vec 无法预先调整大小 - 它只是从大小 1 开始并从那里增长。
3. 对于典型的字符串，它分配的空间比存储最终结果实际所需的空间多，因为&str 通常是 16 字节大小，而 UTF-8 字形簇通常是远不止于此。
4. 在中间 Vec 上进行迭代以获取最终大小是浪费时间，您可以从原始 &str 中获取它。

最重要的是，我什至不认为这个版本是惯用的，因为它 collects 到一个临时的 Vec 以便对其进行迭代，而不是仅仅 collecting 原始迭代器，如你有你的答案的早期版本。此版本修复了问题 #3，使 #4 无关紧要，但不能令人满意地解决问题 #2：

input.graphemes(true).rev().collect()

collect 将FromIterator 用于String，这将是try to use 来自Iterator 实现的Graphemes 的size_hint 的下限。但是，正如我之前提到的，扩展的字形簇可以任意长，因此下限不能大于 1。更糟糕的是，&strs 可能为空，所以 FromIterator<&str> 为 String 不知道任何关于结果大小的内容（以字节为单位）。这段代码只是创建了一个空的String 并对其反复调用push_str。

需要明确的是，这还不错！ String 有一个增长策略，可以保证分期 O(1) 插入，所以如果你有大部分不需要经常重新分配的小字符串，或者你不相信分配成本是一个瓶颈，使用 @987654352如果您发现它更易读且更容易推理，这里的 @ 可能是合理的。

让我们回到你原来的代码。

let mut result = String::with_capacity(input.len());
for gc in input.graphemes(true).rev() {
    result.push_str(gc);
}

这是惯用的。 collect 也是惯用的，但所有collect 所做的基本上都是上述内容，初始容量不太准确。由于collect 没有做你想做的事，所以自己编写代码并不习惯。

还有一个更简洁的迭代器版本，它仍然只进行一次分配。使用extend 方法，它是Extend<&str> 的一部分，用于String：

fn reverse(input: &str) -> String {
    let mut result = String::with_capacity(input.len());
    result.extend(input.graphemes(true).rev());
    result
}

我有一种模糊的感觉，extend 更好，但这两种方式都是编写相同代码的完全惯用方式。你不应该重写它以使用collect，除非你觉得它更好地表达了意图并且你不关心额外的分配。

相关

• Efficiency of flattening and collecting slices