c# - 有没有更好的方法来返回列表中的下一项并从末尾循环到前面？

Question

我有以下不同字符串列表:

“一个”
“乙”
“C”

如果我想要A之后的项目，我得到B。在B之后，我得到C。在C之后，我得到A。目前我有以下代码，但出于某种原因，我觉得有更好的方法去解决这个问题(也许？)。

private string GetNext(IList<string> items, string curr)
{
    if (String.IsNullOrWhitespace(curr))
        return items[0];

    var index = items.IndexOf(curr);
    if (index == -1)
        return items[0];

    return (index + 1 == items.Count) ? items[0] : items[index + 1];
}

我当然也愿意使用类似 LINQ 的方式来做这件事 :)

Answer 1

您拥有的解决方案在功能上是正确的，但它的性能还有一些不足之处。通常，在处理列表样式结构时，您会期望 GetNext 会在 O(1) 时间内返回结果，但此解决方案是 O(N)。

public sealed class WrappingIterator<T> {
  private IList<T> _list;
  private int _index;
  public WrappingIterator<T>(IList<T> list, int index) {
    _list = list;
    _index = index;
  }
  public T GetNext() {
    _index++;
    if (_index == _list.Count) {
      _index = 0;
    }
    return _list[_index];
  }

  public static WrappingIterator<T> CreateAt(IList<T> list, T value) {
    var index = list.IndexOf(value);
    return new WrappingIterator(list, index);
  }
}

这里对 CreateAt 的初始调用是 O(N)，但对 GetNext 的后续调用是 O(1)。

IList<string> list = ...;
var iterator = WrappingIterator<string>.CreateAt(list, "B");
Console.WriteLine(iterator.GetNext());  // Prints C
Console.WriteLine(iterator.GetNext());  // Prints A
Console.WriteLine(iterator.GetNext());  // Prints B