主页 / 问题 / 1203423
Accepted
skyho
Asked:2021-11-12 05:38:03 +0000 UTC2021-11-12 05:38:03 +0000 UTC

如何将值数组拆分为给定数量的组

  • 772

例如，有这个代码

 final List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11);
        final int chunkSize = 3;
        final AtomicInteger counter = new AtomicInteger();

        final Collection<List<Integer>> result = numbers.stream()
                .collect(Collectors.groupingBy(it -> counter.getAndIncrement() / chunkSize))
                .values();

        System.out.println(result);

之类的

 List<String> originalList = Arrays.asList("1", "2", "3",
                "4", "5", "6", "7", "8", "9", "10", "11");

        int partitionSize = 3;

        List<List<String>> partitions = new LinkedList<>();

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

            int min = Math.min(i + partitionSize, originalList.size());
            partitions.add(originalList.subList(i, min ));
        }

        System.out.println(partitions);

但是它们创建了单独的数组，其中放置了给定数量的元素。

但挑战是以这样一种方式划分数组（数组是 N 长度），您可以将原始数组中的元素数量均匀地放入其中。

比如有一个16个文件的文件数组，你需要把原来的数组拆分成3组，然后从这个数组中把一定数量的文件均匀的复​​制到这3组中。例如，在第一组中，将有 5 个元素，在第二组 - 5 个元素中，在第 3 组 - 6 个元素中......


File[] files = sourceFolder.listFiles();

File[] filesFirstArr = divideArray(files, ?,  ?);

    public static File[] divideArray(File[] filesOrigin, int posCopySource , int sizeArrayFiles){

        File[] filesPart = new File[sizeArrayFiles];
        System.arraycopy(filesOrigin, posCopySource, filesPart, 0, sizeArrayFiles);

        return filesPart;
    }

该方法将像这样在循环中调用


File[] files = sourceFolder.listFiles();

for(int i = 0; i < sourceFolder.length(); i += partitionSize){
...
}

我不知道如何计算复制元素位置的值（初始，对于每个新的元素组），我不知道如何计算数组的下一个大小，下一组元素）

谁有想法怎么做？

java

2 个回答

  1. Best Answer

    如何计算数组的下一个大小，下一组元素

    int partitionSize = files.length / numberOfChunks;
int rest = files.length % numberOfChunks;

// Принимаем решение о том, что делать с неполным остатком
boolean includeRest = rest < partitionSize / 2;

// уменьшаем количество итераций на одну, если нужно
int bound = sequence.length - (includeRest ? partitionSize : 0);
for(int i = 0; i < bound; i += partitionSize){
    boolean isLastChunk = (i + partitionSize) >= bound;
    File[] chunk = Arrays.copyOfRange(files, i,
          (isLastChunk && includeRest)
              ? files.length
              : Math.min(files.length, i + partitionSize));
    // ...
}

    在线演示

    • 0

  2. 更新（基于论坛成员建议的解决方案 - vp_arth ）

    
 public static void main(String[] args){

        int partitionSize = 3;

        FileWriteRepository fileWriteRepository = new FileWriteRepositoryImpl();
        String srcFolder = "source";
        File sourceFolder = fileWriteRepository.getPathDirToSource(srcFolder);

        File[] files = sourceFolder.listFiles();

        List<File[]> dividingFiles = runDividingFiles(files, partitionSize);
}

private static List<File[]> runDividingFiles(File[] files, int numberOfChunks){

        List<File[]> resultArr = new ArrayList<>();

        /*количество файлов в одном сегменте*/
        int partitionSize = files.length / numberOfChunks;

        /*получаем размер последнего неполного сегмента*/
        int rest = files.length % numberOfChunks;

        /*получаем половину одного сегмента*/
        int halfOneSegment = partitionSize / 2;

        /*проверяем, нужно ли включать остаток в текущий собираемый массив*/
        boolean includeRest = false;

        if(rest > 0 && rest <= halfOneSegment) includeRest = true;

        /* уменьшаем количество итераций на одну, если нужно*/
        int bound = files.length - (includeRest ? partitionSize : 0);

        /*каждый цикл формируем новый массив файлов, который представляет собой
         * один из сегментов оригинального большого массива*/
        for (int i = 0; i < bound; i += partitionSize) {

            /*вычисляем минимальное значение, между длиной массива и текущим последним
             * индексом создаваемого сегмента*/
            int min = Math.min(files.length, i + partitionSize);

            /* Проверяем, нужно ли включать остаток в последний сегмент*/

            boolean isLastChunk = false;

            if((i + partitionSize) >= bound){
                isLastChunk = true;
            }

            int endIndexRange;

            /*определяем последний индекс файла в текущем создаваемом сегменте*/
            if (includeRest && isLastChunk) {

                endIndexRange = files.length; /*если в сегментах четное число файлов*/
            } else {
                endIndexRange = min;
            }

            File[] filesResult = Arrays.copyOfRange(files, i, endIndexRange);

            resultArr.add(filesResult);
        }

        return resultArr;
    }
    • 0

相关问题