多个数组排列问题

有时候会有需求 从N个数组个各取1个元素组成新数组,问有多少种可能,分别是什么?(这里假设不涉及重复元素和顺序影响)java

相似 [a,b,c,d],[1,2],[A,B,F] 3个数组个各取一个元素组成新数组,共计24种可能,结果分别以下:apache

[a,1,A],[a,1,B],[a,1,F],[a,2,A],[a,2,B],[a,2,F],[b,1,A],[b,1,B]......[d,2,A],[d,2,B],[d,2,F].数组

那如何用程序来完成这个数学问题呢?下面先介绍思路,再用java代码尝试code。测试

每次从3个数组中各取一个元素,取元素的下标初始默认分别是0->(0,0,0),即:第一个数组是"a",第二个数组是"1",第三个数组是"A"; 下一个新数组来源的下标呢?前2个仍是0,第三个加1->(0,0,1),同理,再下一个呢?(0,0,2),再下一个呢? 是0,0,3吗?code

这里须要注意一下,当取元素的数组的下标越界(>=当前数组的长度)时,咱们须要将当前元素的数组的下标置为0,前一个数组的下标加1,因此(0,0,2)后面一个下标应该是(0,1,0);后面规律就同样了。索引

看这个过程像什么呢?是否是相似进制数+1的过程?get

若是是10进制的,2个数组就相似组成两位数,3个数组相似组成3位数。可能性= 10*10*...*10,有多少个数组就有10N种可能数学

若是是2进制的,即,每一个源数组都有2个元素,那可能性=2*2*...*2 有多少个数组就有2N种可能;it

如今数组长度是不规律的,可能性= 每一个数组元素个数乘积,上方示例可能=4*2*3=24种;class

知道了过程和原理开始撸代码吧:

给出的参数就是1个集合,集合的每一个元素是1个数组

List<String[]> originalList

首先须要2个临时变量数组,分别存放原数组取数的下标和原数组的长度

int[] tempIndexArr = new int[originalList.size()];

int[] tempLengthArr = new int[originalList.size()];

须要1个集合存放新组成的数组

List<String[]> combineList = new ArrayList<>();

须要一个是否完成遍历的flag

boolean completeFlag = false;

最核心的逻辑,其实也很简单,就是取下标索引:

int changeIndex = originalList.size() - 1;

boolean isRightIndex = false;

while (!isRightIndex) {

    tempIndexArr[changeIndex] += 1;

    if(tempIndexArr[changeIndex] >= tempLengthArr[changeIndex]) {

        tempIndexArr[changeIndex--] = 0;

    } else {

        isRightIndex = true;

    }

}

取到了下标索引,从原集合中根据下标索引取数组成新数组放到新集合中

String[] newItem = new String[originalList.size()];

for (int i = 0; i < originalList.size() ; i++) {

    newItem[i] = originalList.get(i)[tempIndexArr[i]];

}

combineList.add(newItem);

基本过程就结束了。 不过这里须要增长一个判断 就是是否遍历完了,判断逻辑是 下标都已经到了最大值,完成代码、测试和打印结果以下

package work;

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

import org.apache.commons.lang.StringUtils;

/**
 * @author lixiaoming
 * @date 2019/6/19 16:53
 */
public class CombineTest {

    public static void main(String[] args) {
        List<String[]> originalList = new ArrayList<>();
        originalList.add(new String[] {"a", "b", "c", "d"});
        originalList.add(new String[] {"1", "2"});
        originalList.add(new String[] {"A", "B", "F"});

        List<String[]> combineList = getCombineList(originalList);

        System.out.println("组成的新集合共有" + combineList.size() + "个数组");

        combineList.forEach(s-> {
            System.out.println(StringUtils.join(s, ","));
        });
    }

    private static List<String[]> getCombineList(List<String[]> originalList) {
        int originalSize = originalList.size();
        int[] tempIndexArr = new int[originalSize];
        tempIndexArr[originalSize - 1] = -1;
        int[] tempLengthArr = new int[originalSize];
        for (int i = 0; i < originalSize ; i++) {
            tempLengthArr[i] = originalList.get(i).length;
        }

        List<String[]> combineList = new ArrayList<>();
        boolean completeFlag = false;

        while(!completeFlag) {
            int changeIndex = originalList.size() - 1;
            boolean isRightIndex = false;
            while (!isRightIndex) {
                tempIndexArr[changeIndex] += 1;
                if(tempIndexArr[changeIndex] >= tempLengthArr[changeIndex]) {
                    if(changeIndex == 0) {
                        isRightIndex = true;
                        completeFlag = true;
                    } else {
                        tempIndexArr[changeIndex--] = 0;
                    }
                } else {
                    isRightIndex = true;
                }
            }
            if(isRightIndex && !completeFlag) {
                String[] newItem = new String[originalList.size()];
                for (int i = 0; i < originalList.size() ; i++) {
                    newItem[i] = originalList.get(i)[tempIndexArr[i]];
                }
                combineList.add(newItem);
            }
        }

        return combineList;
    }

}

打印结果以下:

组成的新集合共有24个数组

a,1,A

a,1,B

a,1,F

a,2,A

a,2,B

a,2,F

b,1,A

b,1,B

b,1,F

b,2,A

b,2,B

b,2,F

c,1,A

c,1,B

c,1,F

c,2,A

c,2,B

c,2,F

d,1,A

d,1,B

d,1,F

d,2,A

d,2,B

d,2,F

Process finished with exit code 0

相关文章
相关标签/搜索