001/*
002 * Copyright (C) 2016 The Guava Authors
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017package com.google.common.collect;
018
019import static com.google.common.base.Preconditions.checkNotNull;
020import static com.google.common.collect.CollectPreconditions.checkNonnegative;
021
022import com.google.common.annotations.Beta;
023import com.google.common.annotations.GwtCompatible;
024import java.util.Comparator;
025import java.util.Iterator;
026import java.util.List;
027import java.util.stream.Collector;
028import org.checkerframework.checker.nullness.qual.Nullable;
029
030/**
031 * Provides static methods for working with {@link Comparator} instances. For many other helpful
032 * comparator utilities, see either {@code Comparator} itself (for Java 8+), or {@code
033 * com.google.common.collect.Ordering} (otherwise).
034 *
035 * <h3>Relationship to {@code Ordering}</h3>
036 *
037 * <p>In light of the significant enhancements to {@code Comparator} in Java 8, the overwhelming
038 * majority of usages of {@code Ordering} can be written using only built-in JDK APIs. This class is
039 * intended to "fill the gap" and provide those features of {@code Ordering} not already provided by
040 * the JDK.
041 *
042 * @since 21.0
043 * @author Louis Wasserman
044 */
045@GwtCompatible
046@ElementTypesAreNonnullByDefault
047public final class Comparators {
048  private Comparators() {}
049
050  /**
051   * Returns a new comparator which sorts iterables by comparing corresponding elements pairwise
052   * until a nonzero result is found; imposes "dictionary order." If the end of one iterable is
053   * reached, but not the other, the shorter iterable is considered to be less than the longer one.
054   * For example, a lexicographical natural ordering over integers considers {@code [] < [1] < [1,
055   * 1] < [1, 2] < [2]}.
056   *
057   * <p>Note that {@code Collections.reverseOrder(lexicographical(comparator))} is not equivalent to
058   * {@code lexicographical(Collections.reverseOrder(comparator))} (consider how each would order
059   * {@code [1]} and {@code [1, 1]}).
060   */
061  // Note: 90% of the time we don't add type parameters or wildcards that serve only to "tweak" the
062  // desired return type. However, *nested* generics introduce a special class of problems that we
063  // think tip it over into being worthwhile.
064  public static <T extends @Nullable Object, S extends T> Comparator<Iterable<S>> lexicographical(
065      Comparator<T> comparator) {
066    return new LexicographicalOrdering<S>(checkNotNull(comparator));
067  }
068
069  /**
070   * Returns {@code true} if each element in {@code iterable} after the first is greater than or
071   * equal to the element that preceded it, according to the specified comparator. Note that this is
072   * always true when the iterable has fewer than two elements.
073   */
074  public static <T extends @Nullable Object> boolean isInOrder(
075      Iterable<? extends T> iterable, Comparator<T> comparator) {
076    checkNotNull(comparator);
077    Iterator<? extends T> it = iterable.iterator();
078    if (it.hasNext()) {
079      T prev = it.next();
080      while (it.hasNext()) {
081        T next = it.next();
082        if (comparator.compare(prev, next) > 0) {
083          return false;
084        }
085        prev = next;
086      }
087    }
088    return true;
089  }
090
091  /**
092   * Returns {@code true} if each element in {@code iterable} after the first is <i>strictly</i>
093   * greater than the element that preceded it, according to the specified comparator. Note that
094   * this is always true when the iterable has fewer than two elements.
095   */
096  public static <T extends @Nullable Object> boolean isInStrictOrder(
097      Iterable<? extends T> iterable, Comparator<T> comparator) {
098    checkNotNull(comparator);
099    Iterator<? extends T> it = iterable.iterator();
100    if (it.hasNext()) {
101      T prev = it.next();
102      while (it.hasNext()) {
103        T next = it.next();
104        if (comparator.compare(prev, next) >= 0) {
105          return false;
106        }
107        prev = next;
108      }
109    }
110    return true;
111  }
112
113  /**
114   * Returns a {@code Collector} that returns the {@code k} smallest (relative to the specified
115   * {@code Comparator}) input elements, in ascending order, as an unmodifiable {@code List}. Ties
116   * are broken arbitrarily.
117   *
118   * <p>For example:
119   *
120   * <pre>{@code
121   * Stream.of("foo", "quux", "banana", "elephant")
122   *     .collect(least(2, comparingInt(String::length)))
123   * // returns {"foo", "quux"}
124   * }</pre>
125   *
126   * <p>This {@code Collector} uses O(k) memory and takes expected time O(n) (worst-case O(n log
127   * k)), as opposed to e.g. {@code Stream.sorted(comparator).limit(k)}, which currently takes O(n
128   * log n) time and O(n) space.
129   *
130   * @throws IllegalArgumentException if {@code k < 0}
131   * @since 33.2.0 (available since 22.0 in guava-jre)
132   */
133  @SuppressWarnings({"AndroidJdkLibsChecker", "Java7ApiChecker"})
134  @IgnoreJRERequirement // Users will use this only if they're already using streams.
135  @Beta // TODO: b/288085449 - Remove.
136  public static <T extends @Nullable Object> Collector<T, ?, List<T>> least(
137      int k, Comparator<? super T> comparator) {
138    checkNonnegative(k, "k");
139    checkNotNull(comparator);
140    return Collector.of(
141        () -> TopKSelector.<T>least(k, comparator),
142        TopKSelector::offer,
143        TopKSelector::combine,
144        TopKSelector::topK,
145        Collector.Characteristics.UNORDERED);
146  }
147
148  /**
149   * Returns a {@code Collector} that returns the {@code k} greatest (relative to the specified
150   * {@code Comparator}) input elements, in descending order, as an unmodifiable {@code List}. Ties
151   * are broken arbitrarily.
152   *
153   * <p>For example:
154   *
155   * <pre>{@code
156   * Stream.of("foo", "quux", "banana", "elephant")
157   *     .collect(greatest(2, comparingInt(String::length)))
158   * // returns {"elephant", "banana"}
159   * }</pre>
160   *
161   * <p>This {@code Collector} uses O(k) memory and takes expected time O(n) (worst-case O(n log
162   * k)), as opposed to e.g. {@code Stream.sorted(comparator.reversed()).limit(k)}, which currently
163   * takes O(n log n) time and O(n) space.
164   *
165   * @throws IllegalArgumentException if {@code k < 0}
166   * @since 33.2.0 (available since 22.0 in guava-jre)
167   */
168  @SuppressWarnings({"AndroidJdkLibsChecker", "Java7ApiChecker"})
169  @IgnoreJRERequirement // Users will use this only if they're already using streams.
170  @Beta // TODO: b/288085449 - Remove.
171  public static <T extends @Nullable Object> Collector<T, ?, List<T>> greatest(
172      int k, Comparator<? super T> comparator) {
173    return least(k, comparator.reversed());
174  }
175
176  /**
177   * Returns the minimum of the two values. If the values compare as 0, the first is returned.
178   *
179   * <p>The recommended solution for finding the {@code minimum} of some values depends on the type
180   * of your data and the number of elements you have. Read more in the Guava User Guide article on
181   * <a href="https://github.com/google/guava/wiki/CollectionUtilitiesExplained#comparators">{@code
182   * Comparators}</a>.
183   *
184   * @param a first value to compare, returned if less than or equal to b.
185   * @param b second value to compare.
186   * @throws ClassCastException if the parameters are not <i>mutually comparable</i>.
187   * @since 30.0
188   */
189  public static <T extends Comparable<? super T>> T min(T a, T b) {
190    return (a.compareTo(b) <= 0) ? a : b;
191  }
192
193  /**
194   * Returns the minimum of the two values, according to the given comparator. If the values compare
195   * as equal, the first is returned.
196   *
197   * <p>The recommended solution for finding the {@code minimum} of some values depends on the type
198   * of your data and the number of elements you have. Read more in the Guava User Guide article on
199   * <a href="https://github.com/google/guava/wiki/CollectionUtilitiesExplained#comparators">{@code
200   * Comparators}</a>.
201   *
202   * @param a first value to compare, returned if less than or equal to b
203   * @param b second value to compare.
204   * @throws ClassCastException if the parameters are not <i>mutually comparable</i> using the given
205   *     comparator.
206   * @since 30.0
207   */
208  @ParametricNullness
209  public static <T extends @Nullable Object> T min(
210      @ParametricNullness T a, @ParametricNullness T b, Comparator<? super T> comparator) {
211    return (comparator.compare(a, b) <= 0) ? a : b;
212  }
213
214  /**
215   * Returns the maximum of the two values. If the values compare as 0, the first is returned.
216   *
217   * <p>The recommended solution for finding the {@code maximum} of some values depends on the type
218   * of your data and the number of elements you have. Read more in the Guava User Guide article on
219   * <a href="https://github.com/google/guava/wiki/CollectionUtilitiesExplained#comparators">{@code
220   * Comparators}</a>.
221   *
222   * @param a first value to compare, returned if greater than or equal to b.
223   * @param b second value to compare.
224   * @throws ClassCastException if the parameters are not <i>mutually comparable</i>.
225   * @since 30.0
226   */
227  public static <T extends Comparable<? super T>> T max(T a, T b) {
228    return (a.compareTo(b) >= 0) ? a : b;
229  }
230
231  /**
232   * Returns the maximum of the two values, according to the given comparator. If the values compare
233   * as equal, the first is returned.
234   *
235   * <p>The recommended solution for finding the {@code maximum} of some values depends on the type
236   * of your data and the number of elements you have. Read more in the Guava User Guide article on
237   * <a href="https://github.com/google/guava/wiki/CollectionUtilitiesExplained#comparators">{@code
238   * Comparators}</a>.
239   *
240   * @param a first value to compare, returned if greater than or equal to b.
241   * @param b second value to compare.
242   * @throws ClassCastException if the parameters are not <i>mutually comparable</i> using the given
243   *     comparator.
244   * @since 30.0
245   */
246  @ParametricNullness
247  public static <T extends @Nullable Object> T max(
248      @ParametricNullness T a, @ParametricNullness T b, Comparator<? super T> comparator) {
249    return (comparator.compare(a, b) >= 0) ? a : b;
250  }
251}