org.teiid.query.util.Permutation Maven / Gradle / Ivy

Go to download
/*
 * Copyright Red Hat, Inc. and/or its affiliates
 * and other contributors as indicated by the @author tags and
 * the COPYRIGHT.txt file distributed with this work.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.teiid.query.util;

import java.util.Iterator;
import java.util.NoSuchElementException;

import org.teiid.core.util.ArgCheck;


public class Permutation {

	private Object[] items;

	public Permutation(Object[] items) {
		ArgCheck.isNotNull(items);
		this.items = items;
	}

	/**
	 * Create list of arrays of items, in all possible permutations (that's n! permutations).
	 * @return Iterator where each thing returned by the iterator is a permutation Object[]
	 */
	public Iterator generate() {
		return new PermutationIterator(this.items, this.items.length);
	}

	/**
	 * Create list of arrays of items, in all possible permutations (that's n! permutations).
	 * @return Iterator where each thing returned by the iterator is a permutation Object[] of
	 * length size
	 */
	public Iterator generate(int size) {
		if(size > items.length) {
			throw new IllegalArgumentException("Size is larger than length"); //$NON-NLS-1$
		} else if(size < 0) {
			throw new IllegalArgumentException("Size is negative"); //$NON-NLS-1$
		}

		return new PermutationIterator(this.items, size);
	}

	private static class PermutationIterator implements Iterator {
		// Given state
		private Object[] items;
		private int k;

		// Derived state
		private int n;
		private int[] j;
		private boolean more = true;
		private int[] last;

		private PermutationIterator(Object[] items, int k) {
			this.items = items;
			this.k = k;
			this.n = items.length;

			if(n == 0 || k == 0) {
				more = false;
			} else {
				// Initialize
				j = new int[n];
				j[0] = -1;

				// terminator
				last = new int[n];
				for(int i=n-1; i>=0; i--) {
					last[n-1-i] = i;
				}
			}
		}

		public boolean hasNext() {
			return more;
		}

		public Object[] next() {
			if(! more) {
				throw new NoSuchElementException();
			}

			if(j[0] < 0) {
				for(int i=0; i= 0 && j[i] >= j[i+1]; --i) {}

				if(i >= 0) {
					int least = i+1;
					for(int m=i+2; m j[i]) {
							least = m;
						}
					}
					int t = j[i];
					j[i] = j[least];
					j[least] = t;
					if(k-1 > i) {
						int start = i+1;
						int end = n-1;
						while(start < end) {
							t = j[start];
							j[start++] = j[end];
							j[end--] = t;
						}
						start = k;
						end = n-1;
						while(start < end) {
							t = j[start];
							j[start++] = j[end];
							j[end--] = t;
						}
					}
				}
			}

			// Check for end
			more = false;
			for(int x=0; x