在 Java 中转置 ArrayList<ArrayList<String>>

Question

我有一个方法试图转置包含字符串 ArrayList 的 ArrayList，称为 matrix 和 return 新数组。

我找到了 Transposing Values in Java 2D ArrayList，但它看起来像是用于 Arrays 而不是 ArrayLists。我的二维数组的尺寸未知，可能是矩形或不规则的（但绝不是正方形）。

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

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;
}

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

我确定有更简洁的方法，但这似乎有效。

在 Java 中转置 ArrayList<ArrayList<String>>

Transpose ArrayList<ArrayList<String>> in Java

java

transpose

arraylist