有没有更好的方法来返回列表中的下一个项目并从头到尾循环？

Question

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

"A"
"B"
"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

Answer 2

我想也许你可以换线

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

喜欢的东西

return items[(index + 1) % items.Count];