Leetcode - Range Sum Query 2D - Mutable

My code:

public class NumMatrix {
    int[][] matrix;
    int[][] BIT;
    public NumMatrix(int[][] matrix) {
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
            return;
        }
        this.matrix = matrix;
        BIT = new int[matrix.length][matrix[0].length + 1];
        for (int i = 0; i < BIT.length; i++) {
            for (int j = 1; j < BIT[0].length; j++) {
                updateBIT(i, j - 1, matrix[i][j - 1]);
            }
        }
    }

    public void update(int row, int col, int val) {
        if (this.BIT == null) {
            return;
        }
        int diff = val - matrix[row][col];
        matrix[row][col] = val;
        updateBIT(row, col, diff);
    }

    public int sumRegion(int row1, int col1, int row2, int col2) {
        if (BIT == null) {
            return -1;
        }
        if (col1 == 0) {
            int sum = 0;
            for (int i = row1; i <= row2; i++) {
                sum += getSum(i, col2);
            }
            return sum;
        }
        else {
            int sum = 0;
            for (int i = row1; i <= row2; i++) {
                sum += getSum(i, col2) - getSum(i, col1 - 1);
            }
            return sum;
        }
    }
    
    private void updateBIT(int row, int col, int diff) {
        int index = col + 1;
        while (index < BIT[0].length) {
            BIT[row][index] += diff;
            index += (index & -index);
        }
    }
    
    private int getSum(int row, int col) {
        int index = col + 1;
        int ans = 0;
        while (index > 0) {
            ans += BIT[row][index];
            index -= (index & -index);
        }
        
        return ans;
    }
}


// Your NumMatrix object will be instantiated and called as such:
// NumMatrix numMatrix = new NumMatrix(matrix);
// numMatrix.sumRegion(0, 1, 2, 3);
// numMatrix.update(1, 1, 10);
// numMatrix.sumRegion(1, 2, 3, 4);

這是我自己寫出來(lái)的徐紧。
把 2dmatrix的每一行當(dāng)成獨(dú)立的BIT,然后,來(lái)求暇矫。
構(gòu)建 2d BIT的時(shí)間復(fù)雜度是:O(MN log N)
sum: O(M log N)
update: O(M log N)

然后看了答案的一種寫法,一種沒徹底搞清楚是什么原理择吊,只能意會(huì)李根。

My code:

public class NumMatrix {

    int[][] tree;
    int[][] nums;
    int m;
    int n;
    
    public NumMatrix(int[][] matrix) {
        if (matrix.length == 0 || matrix[0].length == 0) return;
        m = matrix.length;
        n = matrix[0].length;
        tree = new int[m+1][n+1];
        nums = new int[m][n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                update(i, j, matrix[i][j]);
            }
        }
    }

    public void update(int row, int col, int val) {
        if (m == 0 || n == 0) return;
        int delta = val - nums[row][col];
        nums[row][col] = val;
        for (int i = row + 1; i <= m; i += i & (-i)) {
            for (int j = col + 1; j <= n; j += j & (-j)) {
                tree[i][j] += delta;
            }
        }
    }

    public int sumRegion(int row1, int col1, int row2, int col2) {
        if (m == 0 || n == 0) return 0;
        return sum(row2+1, col2+1) + sum(row1, col1) - sum(row1, col2+1) - sum(row2+1, col1);
    }
    
    public int sum(int row, int col) {
        int sum = 0;
        for (int i = row; i > 0; i -= i & (-i)) {
            for (int j = col; j > 0; j -= j & (-j)) {
                sum += tree[i][j];
            }
        }
        return sum;
    }
}

reference:
https://discuss.leetcode.com/topic/30343/java-2d-binary-indexed-tree-solution-clean-and-short-17ms/2

http://www.geeksforgeeks.org/two-dimensional-binary-indexed-tree-or-fenwick-tree/

Time Complexity:
Both updateBIT(x, y, val) function and getSum(x, y) function takes O(log(NM)) time.
Building the 2D BIT takes O(NM log(NM)).
Since in each of the queries we are calling getSum(x, y) function so answering all the Q queries takesO(Q.log(NM)) time.

Hence the overall time complexity of the program is O((NM+Q).log(NM)) where,N = maximum X co-ordinate of the whole matrix.M = maximum Y co-ordinate of the whole matrix.Q = Number of queries.
**Auxiliary Space: **O(NM) to store the BIT and the auxiliary array

Anyway, Good luck, Richardo! -- 09/04/2016

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市几睛,隨后出現(xiàn)的幾起案子房轿,更是在濱河造成了極大的恐慌,老刑警劉巖所森,帶你破解...
    沈念sama閱讀 219,490評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件囱持,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡焕济,警方通過查閱死者的電腦和手機(jī)纷妆,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,581評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)晴弃,“玉大人掩幢,你說(shuō)我怎么就攤上這事逊拍。” “怎么了际邻?”我有些...
    開封第一講書人閱讀 165,830評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵芯丧,是天一觀的道長(zhǎng)。 經(jīng)常有香客問我世曾,道長(zhǎng)注整,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,957評(píng)論 1 295
  • 正文 為了忘掉前任度硝,我火速辦了婚禮肿轨,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘蕊程。我一直安慰自己椒袍,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,974評(píng)論 6 393
  • 文/花漫 我一把揭開白布藻茂。 她就那樣靜靜地躺著驹暑,像睡著了一般。 火紅的嫁衣襯著肌膚如雪辨赐。 梳的紋絲不亂的頭發(fā)上优俘,一...
    開封第一講書人閱讀 51,754評(píng)論 1 307
  • 那天,我揣著相機(jī)與錄音掀序,去河邊找鬼帆焕。 笑死,一個(gè)胖子當(dāng)著我的面吹牛不恭,可吹牛的內(nèi)容都是我干的叶雹。 我是一名探鬼主播,決...
    沈念sama閱讀 40,464評(píng)論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼换吧,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼折晦!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起沾瓦,我...
    開封第一講書人閱讀 39,357評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤满着,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后贯莺,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體风喇,經(jīng)...
    沈念sama閱讀 45,847評(píng)論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,995評(píng)論 3 338
  • 正文 我和宋清朗相戀三年乖篷,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了响驴。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片透且。...
    茶點(diǎn)故事閱讀 40,137評(píng)論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡撕蔼,死狀恐怖豁鲤,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情鲸沮,我是刑警寧澤琳骡,帶...
    沈念sama閱讀 35,819評(píng)論 5 346
  • 正文 年R本政府宣布,位于F島的核電站讼溺,受9級(jí)特大地震影響楣号,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜怒坯,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,482評(píng)論 3 331
  • 文/蒙蒙 一炫狱、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧剔猿,春花似錦视译、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,023評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至汪茧,卻和暖如春椅亚,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背舱污。 一陣腳步聲響...
    開封第一講書人閱讀 33,149評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工呀舔, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人扩灯。 一個(gè)月前我還...
    沈念sama閱讀 48,409評(píng)論 3 373
  • 正文 我出身青樓别威,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親驴剔。 傳聞我的和親對(duì)象是個(gè)殘疾皇子省古,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,086評(píng)論 2 355

推薦閱讀更多精彩內(nèi)容

  • 背景 一年多以前我在知乎上答了有關(guān)LeetCode的問題, 分享了一些自己做題目的經(jīng)驗(yàn)。 張土汪:刷leetcod...
    土汪閱讀 12,747評(píng)論 0 33
  • 通過Jq獲取到的元素是JQ對(duì)象丧失,通過元素的document.get***獲取到的元素是原生對(duì)象豺妓,原生對(duì)象不能使用J...
    likeli閱讀 202評(píng)論 0 0
  • 亂碼問題locale -a 來(lái)顯示當(dāng)前Linux系統(tǒng)支持的所有的語(yǔ)言環(huán)境export LANG=zh_CN.UTF...
    iscona閱讀 839評(píng)論 0 0
  • IPMI的功能 通過web訪問IPMI,你可以實(shí)現(xiàn)對(duì)機(jī)器的操作布讹,linux 下可以通過ipmitool 直接訪問:...
    鐘大發(fā)閱讀 8,696評(píng)論 0 1