转置ArrayList <ArrayList<String> &gt;在Java中

Question

我有一个方法试图转置一个包含字符串ArrayList的ArrayList，称为矩阵并返回新数组。

我在Java 2D ArrayList中找到了Transposing Values ，但它看起来像是Arrays而不是ArrayLists。 我的2D阵列具有未知的尺寸，可以是矩形的，也可以是不规则的（但不是正方形）。

我的想法是读取每个内部数组，并将项附加到传出矩阵的内部数组。

public static ArrayList<ArrayList<String>> transpose (ArrayList<ArrayList<String>> matrixIn){
    ArrayList<ArrayList<String>> matrixOut = new ArrayList<>();
    //for each row in matrix
    for (int r = 0; r < matrixIn.size(); r++){
        ArrayList<String> innerIn = matrixIn.get(r);

        //for each item in that row
        for (int c = 0; c < innerIn.size(); c++){

            //add it to the outgoing matrix

            //get matrixOut current value
            ArrayList<String> matrixOutRow = matrixOut.get(c);
            //add new one
            matrixOutRow.add(innerIn.get(c));
            //reset to matrixOut
            matrixOut.set(c,matrixOutRow);
        }
    }
    return matrixOut;
}

我收到一个“IndexOutOfBoundsException：Index：0，Size：0”错误

        //get matrixOut[v]
        ArrayList<String> matrixOutRow = matrixOut.get(v);

这件事我做错了什么？

Answer 1

假设：每个内部列表都有相同的元素。 这可以帮到你。

    public static List<List<String>> transpose(ArrayList<ArrayList<String>> matrixIn) {
    List<List<String>> matrixOut = new ArrayList<List<String>>();
    if (!matrixIn.isEmpty()) {
        int noOfElementsInList = matrixIn.get(0).size();
        for (int i = 0; i < noOfElementsInList; i++) {
            List<String> col = new ArrayList<String>();
            for (List<String> row : matrixIn) {
                col.add(row.get(i));
            }
            matrixOut.add(col);
        }
    }

    return matrixOut;
}

Answer 2

在这里回答我自己的问题。 这就是我现在正在做的事情：

    public static ArrayList<ArrayList<String>> transpose (ArrayList<ArrayList<String>> matrixIn){
    ArrayList<ArrayList<String>> matrixOut = new ArrayList<>();
    int rowCount = matrixIn.size();
    int colCount = 0;

    //find max width
    for(int i = 0; i < rowCount; i++){
        ArrayList<String> row = matrixIn.get(i);
        int rowSize = row.size();
        if(rowSize > colCount){
            colCount = rowSize;
        }
    }
    //for each row in matrix
    for (int r = 0; r < rowCount; r++){
        ArrayList<String> innerIn = matrixIn.get(r);

        //for each item in that row
        for (int c = 0; c < colCount; c++){

            //add it to the outgoing matrix
            //get matrixOut[c], or create it
            ArrayList<String> matrixOutRow = new ArrayList<>();
            if (r != 0) {
                try{
                    matrixOutRow = matrixOut.get(c);
                }catch(java.lang.IndexOutOfBoundsException e){
                    System.out.println("Transposition error!\n"
                            + "could not get matrixOut at index "
                             + c + " - out of bounds" +e);
                    matrixOutRow.add("");
                }
            }
            //add innerIn[c]
            try{
                matrixOutRow.add(innerIn.get(c));
            }catch (java.lang.IndexOutOfBoundsException e){
                matrixOutRow.add("");
            }

            //reset to matrixOut[c]
            try {
                matrixOut.set(c,matrixOutRow);                
            }catch(java.lang.IndexOutOfBoundsException e){
                matrixOut.add(matrixOutRow);
            }
        }
    }
    return matrixOut;
}

我不能假设光滑的数组，我想仍然返回一个嵌套的ArrayList。 所以现在我只是通过添加“”找到最大尺寸并捕获所有出界错误。

我确信有一种更清洁的方式，但这似乎有效。