【问题标题】:Removing elements from 2 ArrayList in java inside 2 for loops从2个for循环内的java中的2个ArrayList中删除元素
【发布时间】:2015-09-11 04:10:06
【问题描述】:

有 N 个男孩和 N 个女孩。只有一个男孩和一个女孩可以组成一个舞蹈对(即不允许同性舞蹈对)。成对的唯一其他条件是它们的绝对高度差应小于或等于 K。

找出可以组成的最大配对数,以便每个人都有一个独特的伙伴。

输入格式 第一行将包含两个整数 N 和 K。 第二行将包含 N 个整数,即 N 个男孩的身高。 第三行将包含 N 个整数,即 N 个女孩的身高。

约束 1≤N≤105 1≤K≤109 1≤男孩或女孩身高≤109

输出格式 包含最大可能对数的单行。

我的问题是在满足条件后从两个数组中删除元素。 (存储2个数组后)

我使用的第一个算法:

Arrays.sort(ArrBoys);
    Arrays.sort(ArrGirls);
    double count = 0;
    for (int i = 0; i < ArrGirls.length; i++) {
        for (int j = 0; j < ArrBoys.length; j++) {
            if ((int) Math.abs(ArrBoys[j] - ArrGirls[i]) <= k) {
                ArrBoys[j] = 0;
                ArrGirls[i] = 0;
                count++;
                break;
            }
        }
    }
    System.out.println((int) count);

这绝对是错误的,因为我们需要删除那些元素而不是使它们为零,所以我意识到我需要将它们转换为 ArrayLists。 然后我得到了这段代码:

double count = 0;
    int dif = 0;
    for (Integer ArrGirl : ArrGirls) {
        loop1:
        {
            for (Integer ArrBoy : ArrBoys) {
                dif = (int) Math.abs(ArrBoy - ArrGirl);
                if (dif <= k) {
                    System.out.println("we took " + ArrBoy + " from boys with "
                    + ArrGirl + " from girls, thier dif is " + dif);
                    ArrBoys.remove(ArrBoy);
                    ArrGirls.remove(ArrGirl);
                    count++;
                    break loop1;
                }
            }
        }
    }
    System.out.println((int) count);

使用这个算法给了我一个这样的例外:

Exception in thread "main" java.util.ConcurrentModificationException at java.util.ArrayList$Itr.checkForComodification(ArrayList.java:901) at java.util.ArrayList$Itr.next(ArrayList.java:851)

在我搜索了这个异常之后,我发现我应该在使用 remove 方法时使用 Iterator,然后我使用了这段代码:

double count = 0;
    int dif = 0;
    Iterator<Integer> iteB = ArrGirls.iterator();
    Iterator<Integer> iteG = ArrGirls.iterator();
    while (iteG.hasNext()) {
        int value2 = iteG.next();
        while (iteB.hasNext()) {
            int value = iteB.next();
            dif = (int) Math.abs(value - value2);
            if (dif <= k) {
                System.out.println("we took " + value + " from boys with " + value2
                        + " from girls, thier dif is " + dif);
                count++;
                iteB.remove();
                iteG.remove();

            }
        }
    }
    System.out.println((int) count);

这段代码给了我错误的信息,也给了我同样的异常。

这是我第一次使用ArrayList,您有什么想法可以更正这段代码吗?

【问题讨论】:

  • 例如可以使用bipartite graph matching解决。
  • 您需要在删除后中断(或更改您的逻辑),您不能在调用 next() 之前调用 remove

标签: java arrays algorithm arraylist iterator


【解决方案1】:

你必须每次都为女孩重新分配迭代器。这很好用

 int k = 5;
    int count = 0;
    ArrayList<Integer> ArrGirls = new ArrayList<>((List<Integer>) Arrays.asList(new Integer[]{34, 37, 28, 16, 44, 36, 37, 43, 50, 22}));
    ArrayList<Integer> ArrBoys = new ArrayList<>((List<Integer>) Arrays.asList(new Integer[]{13, 28, 41, 10, 14, 27, 41, 27, 23, 37}));
    for (Iterator<Integer> iteG = ArrGirls.iterator(); iteG.hasNext();) {
        {
            Integer ArrGirl = iteG.next();
            for (Iterator<Integer> iteB = ArrBoys.iterator(); iteB.hasNext();) {
                Integer ArrBoy = iteB.next();
                int dif = (int) Math.abs(ArrBoy - ArrGirl);
                if (dif <= k) {
                    System.out.println("we took " + ArrBoy + " from boys with "
                            + ArrGirl + " from girls, thier dif is " + dif);
                    ArrBoys.remove(ArrBoy);
                    ArrGirls.remove(ArrGirl);
                    iteG = ArrGirls.iterator();//You have to reset the value of iterationGirls
                    count++;
                    break;
                }
            }
        }
    }
    System.out.println(count);

【讨论】:

【解决方案2】:
// This helps I guess with little modifications for taking the inputs
public class Boy {

    private int height;

    public int getHeight() {
        return height;
    }

    public void setHeight(int height) {
        this.height = height;
    }

}

public class Girl {

    private int height;

    public int getHeight() {
        return height;
    }

    public void setHeight(int height) {
        this.height = height;
    }

}

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


public class Program {

    public static void main(String[] args) {
        int k=11;
        Random random = new Random();
        List<Boy> boys = new ArrayList<Boy>();
        List<Girl> girls = new ArrayList<Girl>();
        int noOfcombinations = 0;
        for(int i=1;i<=100;i++)
        {
            Boy boy = new Boy();
            boy.setHeight(random.nextInt(109));
            Girl girl = new Girl();
            girl.setHeight(random.nextInt(109));
            boys.add(boy);
            girls.add(girl);
        }
        for (Boy boy : boys) {
            for (Girl girl : girls) {
                if(Math.abs(boy.getHeight()-girl.getHeight())<k)
                {
                    noOfcombinations++;
                }

            }
        }
        System.out.println("No of combinations is "+noOfcombinations);

    }

}

【讨论】:

  • 不要对 OP 的要求给出不同的解决方案。为给定的问题提供解决方案
  • 感谢您的帮助,我需要知道我的代码中的错误。
猜你喜欢
  • 2015-10-06
  • 1970-01-01
  • 2021-09-09
  • 2021-07-04
  • 2021-05-15
  • 2020-06-02
  • 1970-01-01
  • 2019-11-03
  • 1970-01-01
相关资源
最近更新 更多