java - 在两个数组列表之间交替。

Question

我无法让这个东西运行。我不确定到目前为止我所拥有的是否走在正确的轨道上。我不太确定哪里给了我一个越界错误。

以下是说明:

编写一个名为 interleave 的方法，该方法接受两个整数 a1 和 a2 的 ArrayList 作为参数，并将 a2 的元素插入到 a1 的交替索引处。如果列表的长度不等，则较长列表的剩余元素将保留在 a1 的末尾。例如，如果a1存储[10,20,30]，a2存储[4,5,6,7,8]，则调用interleave(a1,a2)；应该更改 a1 以存储 [10, 4, 20, 5, 30, 6, 7, 8]。如果a1存储了[10, 20, 30, 40, 50]，a2存储了[6, 7, 8]，则调用interleave(a1, a2)；将更改 a1 以存储 [10, 6, 20, 7, 30, 8, 40, 50]。

private static void interleave(ArrayList<Integer> a1,
        ArrayList<Integer> a2) {

    int i = a1.size();
    int j = a2.size();

    if (i < j) { // a1 is shorter than a2
        for (int k = 0; k < a1.size(); k++) { // before k passes a1 size
            a1.add(k+1, a2.get(k));
        }

        for (int l = a1.size(); l < a2.size(); l++) {
            a1.add(a1.size(), a2.get(l));
        }

    } else if (i > j) { // a1 is longer than a2
        for (int k = 1; k < a2.size(); k++) {
            a1.add(k+1, a2.get(k));
        }

    } else { // they are equal length
        for (int k = 1; k < a2.size(); k++) {
            a1.add(k+1, a2.get(k));
        }
    }
}

Answer 1

这应该有效

private static void interleave(ArrayList<Integer> a1, ArrayList<Integer> a2) {
    int i = -1;
    for(Integer elem: a2) {
        if(i < a1.size()-1) {
            i += 2;
        } else {
            i += 1;
        }
        a1.add(i, elem);
    }
}

public static void main(String[] args) throws Exception {

    ArrayList<Integer> a1 = new ArrayList<>(Arrays.asList(10, 20, 30));
    ArrayList<Integer> a2 = new ArrayList<>(Arrays.asList(4, 5, 6, 7, 8));

    interleave(a1, a2);
    System.out.println(a1);
}

编辑:我必须承认，这段代码实际上是一个非常糟糕的解决方案，因为对于长列表来说它会非常慢。每次向 a1 添加一个元素时，列表的一大部分都必须移动一个位置。因此，按照“MadProgrammer”的建议，这是一种更好、更快的方法

private static void interleave(ArrayList<Integer> a1, ArrayList<Integer> a2) {
    ArrayList<Integer> r = new ArrayList<>(a1.size() + a2.size());

    for(int i = 0, j = 0; i < a1.size() || j < a2.size(); i++, j++) {
        if(i < a1.size()) r.add(a1.get(i));
        if(j < a2.size()) r.add(a2.get(j));
    }
    a1.clear();
    a1.addAll(r);
}