JAVA组合函数Combination_java combination-程序员宅基地

技术标签: 算法  

JDK,Apache commons math里都没有发现组合函数,因而自己写了一个。

首先定义接口

package math.combination;

import java.util.Collection;
import java.util.List;

/**
 * the interface for combination
 * @author jhren
 *
 * @param <T>
 */
public interface Combinationable<T> {
	/**
	 * get combinations from start to end [start, end]
	 * it equals findCombiantions(start),findCombiantions(start+1), 
	 * ......
	 * findCombinations(end-1), findCombinations(end)
	 */
	public List<List<T>> findCombinations(int start, int end);
	/**
	 * get all combinations
	 * it equals findCombinations(0, size)
	 */
	public List<List<T>> findCombinations();
	
	/**
	 * find a combinations for size
	 * @param size
	 * @return
	 */
	public List<List<T>> findCombinations(int size);
	
	public Collection<T> getElements();
}

第一个个实现,输出的组合会按照类型T升序排列

例如输入为[3,2,1]的全组合为

[]
[3]
[2]
[1]
[2, 3]
[1, 3]
[1, 2]
[1, 2, 3]

package math.combination;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

/**
 * the T should be Comparable the T in one combination list is sort by the
 * Comparable T
 * 
 * @author jhren
 * 
 * @param <T>
 */
public class ComparableCombination<T extends Comparable<? super T>> implements
		Combinationable<T> {

	private Collection<T> elements = new ArrayList<T>();

	public ComparableCombination(Collection<T> elements) {
		this.elements.addAll(elements);

	}

	@Override
	public List<List<T>> findCombinations(int start, int end) {
		if (start < 0) {
			throw new IllegalArgumentException("start can't be negative: "
					+ start);
		}
		if (end > elements.size()) {
			throw new IllegalArgumentException(
					"end can't be larger than elements size: " + end + ">"
							+ elements.size());
		}
		if(start > end) {
			throw new IllegalArgumentException(
					"start can't be larger than end: " + start + ">"
							+ end);
		}
		List<List<T>> result = new ArrayList<List<T>>();

		for (int i = start; i <= end; i++) {
			result.addAll(findCombinations(i));
		}

		return result;
	}

	@Override
	public List<List<T>> findCombinations() {
		return findCombinations(0, elements.size());
	}

	@Override
	public List<List<T>> findCombinations(int size) {
		if (size < 0) {
			throw new IllegalArgumentException("size can't be negative: "
					+ size);
		}
		if (size > elements.size()) {
			throw new IllegalArgumentException(
					"size can't be larger than elements size: " + size + ">"
							+ elements.size());
		}

		List<List<T>> result = new ArrayList<List<T>>();

		if (size == 0) {
			result.add(new ArrayList<T>());
			return result;
		}

		List<List<T>> combinations = findCombinations(size - 1);
		for (List<T> combination : combinations) {
			for (T element : elements) {
				if (combination.contains(element)) {
					continue;
				}

				List<T> list = new ArrayList<T>();

				list.addAll(combination);

				if (list.contains(element))
					continue;

				list.add(element);
				Collections.sort(list);

				if (result.contains(list))
					continue;

				result.add(list);
			}
		}

		return result;

	}

	@Override
	public Collection<T> getElements() {
		return elements;
	}

}

第二个实现,会保持元素的输入顺序

例如输入为[3,2,1]的全组合为

[]
[3]
[2]
[1]
[3, 2]
[3, 1]
[2, 1]
[3, 2, 1]


package math.combination;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;




/**
 * the T in one combination list is sort by the input sequence
 * @author jhren
 *
 * @param <T>
 */
public class CombinationSortByInput<T> implements Combinationable<T> {
	static private class Element<E> implements Comparable<Element<E>> {
		private int rank;
		private E e;

		public E getE() {
			return e;
		}

		private Element(E e, int rank) {
			this.e = e;
			this.rank = rank;
		}

		@Override
		public int compareTo(Element<E> o) {
			return this.rank - o.rank;
		}

	}

	private Combinationable<Element<T>> combination = null;
	public CombinationSortByInput(Collection<T> elements) {
		Collection<Element<T>> elementsInput = convertCollection(elements);
		combination = new ComparableCombination<Element<T>>(elementsInput);
		
	}
	@Override
	public List<List<T>> findCombinations(int start, int end) {
		List<List<Element<T>>> elementsOutput = combination.findCombinations(start, end);
		List<List<T>> output = convertListOfList(elementsOutput);
		return output;
	}

	@Override
	public List<List<T>> findCombinations() {
		return findCombinations(0, combination.getElements().size());
	}

	@Override
	public List<List<T>> findCombinations(int size) {
		return findCombinations(size,size);
	}

	private static <T> Collection<Element<T>> convertCollection(Collection<T> input) {
		Collection<Element<T>> elementCollectionInput = new ArrayList<Element<T>>();
		Iterator<T> iterator = input.iterator();
		int i = 0;
		while(iterator.hasNext()) {
			T t = iterator.next();
			Element<T> element = new Element<T>(t, i++);
			elementCollectionInput.add(element);
		}
		return elementCollectionInput;
	}
	
	private static <T> List<List<T>> convertListOfList(
			List<List<Element<T>>> elementsOutput) {
		List<List<T>> output = new ArrayList<List<T>>();
		for(List<Element<T>> elementList : elementsOutput) {
			List<T> tlist = convertList(elementList);
			output.add(tlist);
		}
		return output;
	}



	private static <T> List<T> convertList(Collection<Element<T>> elementList) {
		List<T> tlist = new ArrayList<T>();
		for(Element<T> element: elementList) {
			T t = element.getE();
			tlist.add(t);
		}
		return tlist;
	}
	@Override
	public Collection<T> getElements() {
		return convertList(combination.getElements());
	}
}

顺便附上TestNG unitestcode

package math.combination;

import java.util.Arrays;
import java.util.List;

import math.combination.Combinationable;
import math.combination.ComparableCombination;

import org.testng.annotations.Test;
import org.testng.annotations.BeforeClass;
import static org.testng.AssertJUnit.assertNotNull;
import static org.testng.AssertJUnit.assertEquals;
import static org.testng.AssertJUnit.assertTrue;
public class TestComparableCombination {
	private Combinationable<Integer> combination= null;
	@Test
	public void testGetElements() {
		assertNotNull(combination.getElements());
		assertEquals(3, combination.getElements().size());
	}
	
	@Test
	public void testFindCombinationsSizeZero() {
		List<List<Integer>> lists = combination.findCombinations(0);
		assertNotNull(lists);
		assertEquals(1, lists.size());
		assertNotNull(lists.get(0));
		assertEquals(0, lists.get(0).size());
	}
	
	@Test (expectedExceptions = IllegalArgumentException.class)
	public void testFindCombinationsSizeMinus() {
		combination.findCombinations(-1);
		
	}
	
	@Test (expectedExceptions = IllegalArgumentException.class)
	public void testFindCombinationsStartMinus() {
		combination.findCombinations(-1,2);
		
	}
	
	@Test (expectedExceptions = IllegalArgumentException.class)
	public void testFindCombinationsSizeOverflow() {
		combination.findCombinations(combination.getElements().size() + 1);
		
	}
	
	@Test (expectedExceptions = IllegalArgumentException.class)
	public void testFindCombinationsEndOverflow() {
		combination.findCombinations(0,combination.getElements().size() + 1);
		
	}
	
	@Test (expectedExceptions = IllegalArgumentException.class)
	public void testFindCombinationsStartEndWrongOrder() {
		combination.findCombinations(1,0);
		
	}
	
	@Test
	public void testFindCombinationsSizeTwo() {
		List<List<Integer>> lists = combination.findCombinations(2);
		assertNotNull(lists);
		assertEquals(3, lists.size());
		for(List<Integer> list: lists) {
			assertEquals(2, list.size());
			int element0 = list.get(0);
			int elements1 = list.get(1);
			assertTrue(element0 < elements1);
		}
	}
	
	@Test
	public void testFindAllCombinations1() {
		List<List<Integer>> lists = combination.findCombinations();
		assertNotNull(lists);
		assertEquals(8, lists.size());
		
		for(List<Integer> list : lists) {
			System.out.println(list);
		}
	}
	
	@Test
	public void testFindAllCombinations2() {
		List<List<Integer>> lists = combination.findCombinations(0, combination.getElements().size());
		assertNotNull(lists);
		assertEquals(8, lists.size());
	}
	
	@Test(timeOut=20000)
	public void testPerformance() {
		combination = new ComparableCombination<Integer>(Arrays.asList(3,2,1,4,5));
		for(int i = 0;i < 100000; i++) {
			combination.findCombinations();
		}
	}

	@BeforeClass
	public void beforeClass() {
		combination = new ComparableCombination<Integer>(Arrays.asList(3,2,1));
	}

}

package math.combination;

import java.util.Arrays;
import java.util.List;

import math.combination.CombinationSortByInput;
import math.combination.Combinationable;

import org.testng.annotations.Test;
import org.testng.annotations.BeforeClass;
import static org.testng.AssertJUnit.assertNotNull;
import static org.testng.AssertJUnit.assertEquals;
import static org.testng.AssertJUnit.assertTrue;
public class TestCombinationSortByInput {
	private Combinationable<Integer> combination= null;
	@Test
	public void testGetElements() {
		assertNotNull(combination.getElements());
		assertEquals(3, combination.getElements().size());
	}
	
	@Test
	public void testFindCombinationsSizeZero() {
		List<List<Integer>> lists = combination.findCombinations(0);
		assertNotNull(lists);
		assertEquals(1, lists.size());
		assertNotNull(lists.get(0));
		assertEquals(0, lists.get(0).size());
	}
	
	@Test (expectedExceptions = IllegalArgumentException.class)
	public void testFindCombinationsSizeMinus() {
		combination.findCombinations(-1);
		
	}
	
	@Test (expectedExceptions = IllegalArgumentException.class)
	public void testFindCombinationsSizeOverflow() {
		combination.findCombinations(combination.getElements().size() + 1);
		
	}
	
	@Test
	public void testFindCombinationsSizeTwo() {
		List<List<Integer>> lists = combination.findCombinations(2);
		assertNotNull(lists);
		assertEquals(3, lists.size());
		for(List<Integer> list: lists) {
			assertEquals(2, list.size());
			int element0 = list.get(0);
			int elements1 = list.get(1);
			assertTrue(element0 > elements1);
		}
	}
	
	@Test
	public void testFindAllCombinations1() {
		List<List<Integer>> lists = combination.findCombinations();
		assertNotNull(lists);
		assertEquals(8, lists.size());
		
		for(List<Integer> list : lists) {
			System.out.println(list);
		}
	}
	
	@Test
	public void testFindAllCombinations2() {
		List<List<Integer>> lists = combination.findCombinations(0, combination.getElements().size());
		assertNotNull(lists);
		assertEquals(8, lists.size());
	}

	@BeforeClass
	public void beforeClass() {
		combination = new CombinationSortByInput<Integer>(Arrays.asList(3,2,1));
	}
	
	@Test(timeOut=20000)
	public void testPerformance() {
		combination = new CombinationSortByInput<Integer>(Arrays.asList(3,2,1,4,5));
		for(int i = 0;i < 100000; i++) {
			combination.findCombinations();
		}
	}

}


版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/u010157717/article/details/8811923

智能推荐

使用nginx解决浏览器跨域问题_nginx不停的xhr-程序员宅基地

文章浏览阅读1k次。通过使用ajax方法跨域请求是浏览器所不允许的,浏览器出于安全考虑是禁止的。警告信息如下:不过jQuery对跨域问题也有解决方案,使用jsonp的方式解决,方法如下:$.ajax({ async:false, url: 'http://www.mysite.com/demo.do', // 跨域URL ty..._nginx不停的xhr

在 Oracle 中配置 extproc 以访问 ST_Geometry-程序员宅基地

文章浏览阅读2k次。关于在 Oracle 中配置 extproc 以访问 ST_Geometry,也就是我们所说的 使用空间SQL 的方法,官方文档链接如下。http://desktop.arcgis.com/zh-cn/arcmap/latest/manage-data/gdbs-in-oracle/configure-oracle-extproc.htm其实简单总结一下,主要就分为以下几个步骤。..._extproc

Linux C++ gbk转为utf-8_linux c++ gbk->utf8-程序员宅基地

文章浏览阅读1.5w次。linux下没有上面的两个函数,需要使用函数 mbstowcs和wcstombsmbstowcs将多字节编码转换为宽字节编码wcstombs将宽字节编码转换为多字节编码这两个函数,转换过程中受到系统编码类型的影响,需要通过设置来设定转换前和转换后的编码类型。通过函数setlocale进行系统编码的设置。linux下输入命名locale -a查看系统支持的编码_linux c++ gbk->utf8

IMP-00009: 导出文件异常结束-程序员宅基地

文章浏览阅读750次。今天准备从生产库向测试库进行数据导入,结果在imp导入的时候遇到“ IMP-00009:导出文件异常结束” 错误,google一下,发现可能有如下原因导致imp的数据太大,没有写buffer和commit两个数据库字符集不同从低版本exp的dmp文件,向高版本imp导出的dmp文件出错传输dmp文件时,文件损坏解决办法:imp时指定..._imp-00009导出文件异常结束

python程序员需要深入掌握的技能_Python用数据说明程序员需要掌握的技能-程序员宅基地

文章浏览阅读143次。当下是一个大数据的时代,各个行业都离不开数据的支持。因此,网络爬虫就应运而生。网络爬虫当下最为火热的是Python,Python开发爬虫相对简单,而且功能库相当完善,力压众多开发语言。本次教程我们爬取前程无忧的招聘信息来分析Python程序员需要掌握那些编程技术。首先在谷歌浏览器打开前程无忧的首页,按F12打开浏览器的开发者工具。浏览器开发者工具是用于捕捉网站的请求信息,通过分析请求信息可以了解请..._初级python程序员能力要求

Spring @Service生成bean名称的规则(当类的名字是以两个或以上的大写字母开头的话,bean的名字会与类名保持一致)_@service beanname-程序员宅基地

文章浏览阅读7.6k次,点赞2次,收藏6次。@Service标注的bean,类名:ABDemoService查看源码后发现,原来是经过一个特殊处理:当类的名字是以两个或以上的大写字母开头的话,bean的名字会与类名保持一致public class AnnotationBeanNameGenerator implements BeanNameGenerator { private static final String C..._@service beanname

随便推点

二叉树的各种创建方法_二叉树的建立-程序员宅基地

文章浏览阅读6.9w次,点赞73次,收藏463次。1.前序创建#include&lt;stdio.h&gt;#include&lt;string.h&gt;#include&lt;stdlib.h&gt;#include&lt;malloc.h&gt;#include&lt;iostream&gt;#include&lt;stack&gt;#include&lt;queue&gt;using namespace std;typed_二叉树的建立

解决asp.net导出excel时中文文件名乱码_asp.net utf8 导出中文字符乱码-程序员宅基地

文章浏览阅读7.1k次。在Asp.net上使用Excel导出功能,如果文件名出现中文,便会以乱码视之。 解决方法: fileName = HttpUtility.UrlEncode(fileName, System.Text.Encoding.UTF8);_asp.net utf8 导出中文字符乱码

笔记-编译原理-实验一-词法分析器设计_对pl/0作以下修改扩充。增加单词-程序员宅基地

文章浏览阅读2.1k次,点赞4次,收藏23次。第一次实验 词法分析实验报告设计思想词法分析的主要任务是根据文法的词汇表以及对应约定的编码进行一定的识别,找出文件中所有的合法的单词,并给出一定的信息作为最后的结果,用于后续语法分析程序的使用;本实验针对 PL/0 语言 的文法、词汇表编写一个词法分析程序,对于每个单词根据词汇表输出: (单词种类, 单词的值) 二元对。词汇表:种别编码单词符号助记符0beginb..._对pl/0作以下修改扩充。增加单词

android adb shell 权限,android adb shell权限被拒绝-程序员宅基地

文章浏览阅读773次。我在使用adb.exe时遇到了麻烦.我想使用与bash相同的adb.exe shell提示符,所以我决定更改默认的bash二进制文件(当然二进制文件是交叉编译的,一切都很完美)更改bash二进制文件遵循以下顺序> adb remount> adb push bash / system / bin /> adb shell> cd / system / bin> chm..._adb shell mv 权限

投影仪-相机标定_相机-投影仪标定-程序员宅基地

文章浏览阅读6.8k次,点赞12次,收藏125次。1. 单目相机标定引言相机标定已经研究多年,标定的算法可以分为基于摄影测量的标定和自标定。其中,应用最为广泛的还是张正友标定法。这是一种简单灵活、高鲁棒性、低成本的相机标定算法。仅需要一台相机和一块平面标定板构建相机标定系统,在标定过程中,相机拍摄多个角度下(至少两个角度,推荐10~20个角度)的标定板图像(相机和标定板都可以移动),即可对相机的内外参数进行标定。下面介绍张氏标定法(以下也这么称呼)的原理。原理相机模型和单应矩阵相机标定,就是对相机的内外参数进行计算的过程,从而得到物体到图像的投影_相机-投影仪标定

Wayland架构、渲染、硬件支持-程序员宅基地

文章浏览阅读2.2k次。文章目录Wayland 架构Wayland 渲染Wayland的 硬件支持简 述: 翻译一篇关于和 wayland 有关的技术文章, 其英文标题为Wayland Architecture .Wayland 架构若是想要更好的理解 Wayland 架构及其与 X (X11 or X Window System) 结构;一种很好的方法是将事件从输入设备就开始跟踪, 查看期间所有的屏幕上出现的变化。这就是我们现在对 X 的理解。 内核是从一个输入设备中获取一个事件,并通过 evdev 输入_wayland

推荐文章

热门文章

相关标签