Java中合并查找二维数组最高平均行与列的方法

public record HighestAverageResult(int highestRowIndex, int highestColIndex) {
    // 可以添加自定义方法，例如用于格式化输出
    @Override
    public String toString() {
        return "R" + (highestRowIndex + 1) + "C" + (highestColIndex + 1);
    }

    public String getRowString() {
        return "R" + (highestRowIndex + 1);
    }

    public String getColumnString() {
        return "C" + (highestColIndex + 1);
    }
}

import java.util.Arrays;

// 定义用于返回结果的Record类型
public record HighestAverageResult(int highestRowIndex, int highestColIndex) {
    // 默认的访问器方法已由Record自动生成

    // 提供一个自定义的toString方法，以符合R3C3的输出格式
    @Override
    public String toString() {
        // 索引通常从0开始，但输出要求R3、C3，所以需要加1
        return "R" + (highestRowIndex + 1) + ", C" + (highestColIndex + 1);
    }

    // 可以提供单独的格式化方法
    public String toCombinedString() {
        return "R" + (highestRowIndex + 1) + "C" + (highestColIndex + 1);
    }
}

public class ArrayAverageFinder {

    /**
     * 查找二维数组中具有最高平均值的行和列的索引。
     *
     * @param grid 一个二维的double类型数组。
     * @return 包含最高平均行索引和最高平均列索引的HighestAverageResult对象。
     *         如果输入数组为空或不规则，返回-1作为索引。
     */
    public static HighestAverageResult findHighestAverageRowAndColumn(double[][] grid) {
        // 处理空数组或不规则数组的情况
        if (grid == null || grid.length == 0 || grid[0].length == 0) {
            System.out.println("Warning: Input grid is empty or invalid.");
            return new HighestAverageResult(-1, -1); // 返回无效索引
        }

        int numRows = grid.length;
        int numCols = grid[0].length; // 假设所有行的列数相同

        // --- 查找最高平均行 ---
        double maxRowAverage = Double.MIN_VALUE; // 初始化为最小值，确保任何有效平均值都能被更新
        int highestRowIndex = -1;

        for (int i = 0; i < numRows; i++) {
            double currentRowSum = 0;
            // 确保行不为空
            if (grid[i].length == 0) {
                System.out.println("Warning: Row " + i + " is empty, skipping.");
                continue;
            }
            for (int j = 0; j < grid[i].length; j++) {
                currentRowSum += grid[i][j];
            }
            double currentRowAverage = currentRowSum / grid[i].length;

            if (currentRowAverage > maxRowAverage) {
                maxRowAverage = currentRowAverage;
                highestRowIndex = i;
            }
            // System.out.println("Average of row " + (i + 1) + " = " + currentRowAverage); // 调试输出
        }
        // System.out.println("Highest average row is R" + (highestRowIndex + 1) + " with average " + maxRowAverage); // 调试输出

        // --- 查找最高平均列 ---
        double maxColAverage = Double.MIN_VALUE; // 初始化为最小值
        int highestColIndex = -1;

        for (int j = 0; j < numCols; j++) { // 遍历每一列
            double currentColSum = 0;
            for (int i = 0; i < numRows; i++) { // 遍历当前列的所有行
                currentColSum += grid[i][j];
            }
            double currentColAverage = currentColSum / numRows; // 列平均值是列和除以行数

            if (currentColAverage > maxColAverage) {
                maxColAverage = currentColAverage;
                highestColIndex = j;
            }
            // System.out.println("Average of column " + (j + 1) + " = " + currentColAverage); // 调试输出
        }
        // System.out.println("Highest average column is C" + (highestColIndex + 1) + " with average " + maxColAverage); // 调试输出

        return new HighestAverageResult(highestRowIndex, highestColIndex);
    }

    public static void main(String[] args) {
        double[][] matrix = {
            {1.0, 2.0, 3.0},
            {4.0, 5.0, 6.0},
            {7.0, 8.0, 9.0}
        };

        System.out.println("Testing with matrix:");
        for (double[] row : matrix) {
            System.out.println(Arrays.toString(row));
        }

        HighestAverageResult result = findHighestAverageRowAndColumn(matrix);
        System.out.println("\nResult: " + result); // 使用Record的toString方法
        System.out.println("Combined String: " + result.toCombinedString()); // 使用自定义方法
        System.out.println("Highest average row: " + result.getRowString());
        System.out.println("Highest average column: " + result.getColumnString());

        double[][] anotherMatrix = {
            {10.0, 1.0, 2.0},
            {3.0, 40.0, 5.0},
            {6.0, 7.0, 80.0}
        };

        System.out.println("\nTesting with another matrix:");
        for (double[] row : anotherMatrix) {
            System.out.println(Arrays.toString(row));
        }

        HighestAverageResult result2 = findHighestAverageRowAndColumn(anotherMatrix);
        System.out.println("\nResult: " + result2);
        System.out.println("Combined String: " + result2.toCombinedString());
    }
}