如何编写可在列表和数组上工作的方法?

时间:2012-06-05 16:10:53

标签: java arrays list

我有一个看起来像这样的方法:

void foo (List<String> list, ...) {
  ...
  for (String s : list) { // this is the only place where `list` is used
    ...
  }
  ...
}

如果我用List<String> list替换String[] list,完全相同的代码将起作用,但是,为了避免意大利面条代码,我保留单一方法,当我需要在数组{{1}上调用它时我这样做:a

我想知道这是不是正确的方法。

具体来说,

  • foo(Arrays.asList(a))的开销是多少?
  • 有没有办法编写一个接受数组和列表的方法,就像Arrays.asList()循环一样?

谢谢!

5 个答案:

答案 0 :(得分:6)

Arrays.asList()开销很小。没有真正的方法可以为Listarrays实现一种方法。

但您可以执行以下操作:

void foo (List<String> list, ...) {
  ...
  for (String s : list) { // this is the only place where *list* is used
    ...
  }
  ...
}

void foo (String[] arr, ...) {
  if ( arr != null ) {
      foo(Arrays.asList(arr),...);
  }
}

答案 1 :(得分:2)

来自openjdk的源代码,Arrays.asList

public static <T> List<T> asList(T... a) {
   return new ArrayList<>(a);
}

此外:

ArrayList(E[] array) {
   if (array==null)
      throw new NullPointerException();
   a = array;
}

所以基本上所有在分配中都会发生,因此开销应该可以忽略不计。

答案 2 :(得分:1)

开销是将数组转换为列表 - 如何它将依赖于实现,它只需要履行合同。

IMO如果你担心潜在的运行时间开销,你应该写两个方法:这就是Java的本质;方法有类型签名,必须遵守它们。

答案 3 :(得分:1)

请避免这种情况我只是使用并允许列表,集合和地图(like Joshua Bloch told us)。无法合并两种“集合类型”。

另一种方法是使用guava(Iterators / Iteratables)。因此,您可以在没有深层副本的情况下对集合进行迭代。

答案 4 :(得分:1)

好问题。

这是一种非常常见的情况,通常通过编写两种不同的方法来处理。然而,代码重复确实是一个坏主意,每当您发现自己重复代码时,您应该开始寻找机会来更好地考虑您的代码。 (正如你现在正在做的那样!)

现在,如果你查看java.util.Arrays的来源,你会注意到Arrays.asList重新创建一个私有内部类Arrays.ArrayList的实例,它只是普通数组的一个薄包装器,并且委托所有相关的方法调用。 (这被称为数据结构的投影或视图。)因此产生的开销是微不足道的(除非您努力提取性能的最后一点),在我看来,你应该继续使用这种方法而不用担心关于表现。


我个人使用的解决方案如下。

我的个人工具中有一个名为RichIterable的类。正如名称所示,该类包含Iterable,并提供了一些尚未出现的其他有用方法。该类还有一个工厂方法,可以从数组中创建RichIterable。这是类定义。

public class RichIterable<A> implements Iterable<A> {
  private Iterable<A> xs;

  private RichIterable(Iterable<A> xs) {
    this.xs = xs;
  }

  public static <A> RichIterable<A> from(Iterable<A> xs) {
    if (xs instanceof RichIterable) {
      return (RichIterable<A>) xs;
    } else {
      return new RichIterable<A>(xs);
    }
  }

  public static <A> RichIterable<A> from(final Enumeration<A> xs) {
    Iterable<A> iterable = new Iterable<A>() {
      @Override
      public Iterator<A> iterator() {
        return new Iterator<A>() {
          @Override
          public boolean hasNext() {
            return xs.hasMoreElements();
          }

          @Override
          public A next() {
            return xs.nextElement();
          }

          @Override
          public void remove() {
            throw new UnsupportedOperationException(
              "Cannot remove an element from an enumeration.");
          }
        };
      }
    };
    return RichIterable.from(iterable);
  }

  public static <A> RichIterable<A> from(final A[] xs) {
    Iterable<A> iterable = new Iterable<A>() {
      @Override
      public Iterator<A> iterator() {
        return new Iterator<A>() {
          private int i = 0;

          @Override
          public boolean hasNext() {
            return i < xs.length;
          }

          @Override
          public A next() {
            A x = xs[i];
            i++;
            return x;
          }

          @Override
          public void remove() {
            throw new UnsupportedOperationException(
              "Cannot remove an element from an array.");
          }
        };
      }
    };
    return RichIterable.from(iterable);
  }

  public boolean isEmpty() {
    if (xs instanceof Collection) {
      return ((Collection) xs).isEmpty();
    }
    for (A x : xs) {
      return false;
    }
    return true;
  }

  public int size() {
    if (xs instanceof Collection) {
      return ((Collection) xs).size();
    }
    int size = 0;
    for (A x : xs) {
      size++;
    }
    return size;
  }

  public ArrayList<A> toArrayList() {
    ArrayList<A> ys = new ArrayList<A>();
    for (A x : xs) {
      ys.add(x);
    }
    return ys;
  }

  public <B> RichIterable<B> map(F1<A, B> f) {
    List<B> ys = new ArrayList<B>();
    for (A x : xs) {
      ys.add(f.apply(x));
    }
    return RichIterable.from(ys);
  }

  public RichIterable<A> filter(F1<A, Boolean> pred) {
    List<A> ys = new ArrayList<A>();
    Arrays.asList();
    for (A x : xs) {
      if (pred.apply(x)) {
        ys.add(x);
      }
    }
    return RichIterable.from(ys);
  }

  public boolean exists(F1<A, Boolean> pred) {
    for (A x : xs) {
      if (pred.apply(x)) {
        return true;
      }
    }
    return false;
  }

  public boolean forall(F1<A, Boolean> pred) {
    for (A x : xs) {
      if (!pred.apply(x)) {
        return false;
      }
    }
    return true;
  }

  public Maybe<A> find(F1<A, Boolean> pred) {
    for (A x : xs) {
      if (pred.apply(x)) {
        return Just.of(x);
      }
    }
    return Nothing.value();
  }

  public String mkString(String beg, String sep, String end) {
    Iterator<A> i = xs.iterator();
    if (!i.hasNext()) {
      return beg + end;
    }
    StringBuilder sb = new StringBuilder();
    sb.append(beg);
    while (true) {
      A e = i.next();
      sb.append(e.toString());
      if (!i.hasNext()) {
        return sb.append(end).toString();
      }
      sb.append(sep);
    }
  }

  public String mkString(String sep) {
    return mkString("", sep, "");
  }

  public String mkString() {
    return this.mkString(", ");
  }

  public Iterable<A> getRaw() {
    return xs;
  }

  @Override
  public Iterator<A> iterator() {
    return xs.iterator();
  }
}