3484. Design Spreadsheet

A spreadsheet is a grid with 26 columns (labeled from 'A' to 'Z') and a given number of rows. Each cell in the spreadsheet can hold an integer value between 0 and 105.

Implement the Spreadsheet class:

• Spreadsheet(int rows) Initializes a spreadsheet with 26 columns (labeled 'A' to 'Z') and the specified number of rows. All cells are initially set to 0.
• void setCell(String cell, int value) Sets the value of the specified cell. The cell reference is provided in the format "AX" (e.g., "A1", "B10"), where the letter represents the column (from 'A' to 'Z') and the number represents a 1-indexed row.
• void resetCell(String cell) Resets the specified cell to 0.
• int getValue(String formula) Evaluates a formula of the form "=X+Y", where X and Y are either cell references or non-negative integers, and returns the computed sum.

Note: If getValue references a cell that has not been explicitly set using setCell, its value is considered 0.

c++

class Spreadsheet {
    map<pair<int,int>,int> cells;
    pair<int,int> parse(string cell) {
        int row = 0, col = cell[0] - 'A';
        for(int i = 1; i < cell.size(); i++) row = row * 10 + cell[i] - '0';
        return {row,col};
    }
public:
    Spreadsheet(int rows) {}

    void setCell(string cell, int value) {
        cells[parse(cell)] = value;
    }

    void resetCell(string cell) {
        cells.erase(parse(cell));
    }

    int getValue(string formula) {
        string c1 = "", c2 = "";
        while(formula.back() != '+') {
            c2.push_back(formula.back());
            formula.pop_back();
        }
        formula.pop_back();
        while(formula.back() != '=') {
            c1.push_back(formula.back());
            formula.pop_back();
        }
        reverse(begin(c1), end(c1));
        reverse(begin(c2), end(c2));
        int res = 0;
        if(isdigit(c1[0])) {
            res += stoi(c1);
        } else res += cells[parse(c1)];

        if(isdigit(c2[0])) {
            res += stoi(c2);
        } else res += cells[parse(c2)];

        return res;
    }
};

/**
 * Your Spreadsheet object will be instantiated and called as such:
 * Spreadsheet* obj = new Spreadsheet(rows);
 * obj->setCell(cell,value);
 * obj->resetCell(cell);
 * int param_3 = obj->getValue(formula);
 */