com.github.dakusui.enumerator.HomogeniousCombinator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of enumerator Show documentation

A mathematical enumeration library for Java

There is a newer version: 0.6.0

package com.github.dakusui.enumerator;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

public class HomogeniousCombinator extends Combinator {

	private static void hdiv(CDivResult result, long index, int n, int k) {
		int q = 0;
		for (int nn = n; nn > 0; nn--) {
			long nnHk_1 = nHk(nn, k-1); 
			if (index < nnHk_1) {
				result.mod = index;
				result.quotient = q;
				break;
			}
			index -= nnHk_1;
			q++;
		}
	}
	
	private static int[] index2locator(long index, int n, int k) {
		int[] ret = new int[k];
		CDivResult result = new CDivResult();
		for (int i = 0; i < k; i++) {
			hdiv(result, index, n, k - i);
			index = result.mod;
			ret[i] = result.quotient;
			n -= result.quotient;
		}
		return ret;
	}

	public HomogeniousCombinator(List list, int k) {
		super(list, k);
	}

	@Override
	protected List get_Protected(long index) {
		List ret = new LinkedList();
		List tmp = new ArrayList(this.items);
		int[] locator = index2locator(index, items.size(), k);
		for (int i : locator) {
			ret.add(tmp.get(i));
			tmp = tmp.subList(i, tmp.size());
		}
		return ret;
	}
	
	@Override
	public long size() {
		return nHk(this.items.size(), this.k);
	}
}