设计 Excel 求和公式

/*
 * Leetcode 631: 设计 Excel 求和公式
 *
 * 题目思路：
 * 1. 我们需要设计一个 Excel 类，该类可以支持设置单元格的值和计算某个单元格的和。
 * 2. 使用一个二维数组来表示 Excel 表格。
 * 3. 提供 set 方法来设置单元格的值。
 * 4. 提供 sum 方法来计算某个单元格的和，可以传入单个单元格或范围。
 * 5. 使用哈希表来存储每个单元格的求和公式。
 */
 
import java.util.HashMap;
import java.util.Map;
import java.util.stream.IntStream;
 
class Excel {
    private int[][] cells;
    private Map<String, String[]> formulaMap;
 
    public Excel(int height, char width) {
        cells = new int[height + 1][width - 'A' + 1];
        formulaMap = new HashMap<>();
    }
 
    public void set(int row, char column, int val) {
        cells[row][column - 'A'] = val;
        formulaMap.remove(row + "_" + column); // 移除该单元格的公式
    }
 
    public int get(int row, char column) {
        return cells[row][column - 'A'];
    }
 
    public int sum(int row, char column, String[] strs) {
        int sum = IntStream.of(strs).mapToInt(str -> {
            if (str.contains(":")) { // 处理范围
                String[] range = str.split(":");
                int[] start = parsePosition(range[0]);
                int[] end = parsePosition(range[1]);
                return IntStream.rangeClosed(start[0], end[0]).flatMap(r ->
                    IntStream.rangeClosed(start[1], end[1]).map(c -> cells[r][c])
                ).sum();
            } else { // 处理单个单元格
                int[] pos = parsePosition(str);
                return cells[pos[0]][pos[1]];
            }
        }).sum();
        formulaMap.put(row + "_" + column, strs);
        cells[row][column - 'A'] = sum;
        return sum;
    }
 
    private int[] parsePosition(String str) {
        int row = Integer.parseInt(str.substring(1));
        int column = str.charAt(0) - 'A';
        return new int[]{row, column};
    }
}