【ING】dfs總結

https://leetcode.com/tag/depth-first-search/

https://leetcode.com/problems/longest-increasing-path-in-a-matrix/

class Solution {
    int dfs(vector<vector<int>>& matrix, vector<vector<int> > & visited, int x, int y) {
        if(visited[x][y] != -1)
            return visited[x][y];
        int res = 1;
        //left
        if(y > 0 && matrix[x][y-1] > matrix[x][y])
            res = max(res, dfs(matrix, visited, x, y-1) + 1) ;
        //right
        if(y < matrix[0].size() - 1 && matrix[x][y+1] > matrix[x][y])
            res = max(res, dfs(matrix, visited, x, y+1) + 1);
        //up
        if(x > 0 && matrix[x-1][y] > matrix[x][y])
            res = max(res, dfs(matrix, visited, x-1, y) + 1);
        //down
        if(x < matrix.size() - 1 && matrix[x+1][y] > matrix[x][y])
            res = max(res, dfs(matrix, visited, x+1, y) + 1);
        visited[x][y] = res;
        return res;
    }
public:
    int longestIncreasingPath(vector<vector<int>>& matrix) {
        int m = matrix.size();
        if(m == 0) return 0;
        int n = matrix[0].size();
        if(n == 0) return 0;
        
        vector<vector<int> > visited(m,vector<int>(n, -1));
        
        int res = 0;
        for(int i = 0; i < m; i++)
            for(int j = 0; j < n; j++)
                res = max(res, dfs(matrix, visited, i, j));
        return res;
    }
};

https://leetcode.com/problems/maximum-depth-of-binary-tree/
遞歸版本:

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(root == NULL) return 0;
        return max(maxDepth(root->left),maxDepth(root->right)) + 1;
    }
};

非遞歸版本(bfs):

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(root == NULL) return 0;
        queue<TreeNode *> nodeQueue;
        int res = 0;
        nodeQueue.push(root);
        while(!nodeQueue.empty()) {
            res++;
            int tmpsize= nodeQueue.size();
            for(int i = 0; i < tmpsize; i++) {
                TreeNode * temp = nodeQueue.front();
                nodeQueue.pop();
                //cout<<"res="<<res<<"; node="<<temp->val<<endl;
                if(temp->left != NULL)  nodeQueue.push(temp->left);
                if(temp->right != NULL) nodeQueue.push(temp->right);
            }
        }
        return res;
    }
};

same tree
https://leetcode.com/problems/same-tree/

class Solution {
public:
    bool isSameTree(TreeNode* p, TreeNode* q) {
        if(p == NULL && q == NULL) return true;
        if(p == NULL || q == NULL) return false;
        if(p->val != q->val) return false;
        return isSameTree(p->left,q->left) && isSameTree(p->right,q->right);
    }
};

https://leetcode.com/problems/balanced-binary-tree/
這個方法感覺不好,在不停的進行重復的遍歷:

class Solution {
    int getHeight(TreeNode * root) {
        if(root == NULL) return 0;
        return max(getHeight(root->left),getHeight(root->right)) + 1;
    }
public:
    bool isBalanced(TreeNode* root) {
        if(root == NULL) return true;
        int l = getHeight(root->left);
        int r = getHeight(root->right);
        if(abs(l-r) <= 1 && isBalanced(root->left) && isBalanced(root->right)) return true;
        return false;
    }
};

https://leetcode.com/discuss/22898/the-bottom-up-o-n-solution-would-be-better

class solution {
public:
int dfsHeight (TreeNode *root) {
        if (root == NULL) return 0;

        int leftHeight = dfsHeight (root -> left);
        if (leftHeight == -1) return -1;
        int rightHeight = dfsHeight (root -> right);
        if (rightHeight == -1) return -1;

        if (abs(leftHeight - rightHeight) > 1)  return -1;
        return max (leftHeight, rightHeight) + 1;
    }
    bool isBalanced(TreeNode *root) {
        return dfsHeight (root) != -1;
    }
};

https://leetcode.com/problems/number-of-islands/
dfs, bfs, union find

dfs:

class Solution {
    void dfs(vector<vector<char>> &grid, vector<vector<bool>> &visited, int x, int y) {
        if(visited[x][y]) 
            return;
        visited[x][y] = true;
        //up
        if(x > 0 && grid[x-1][y] == '1')
            dfs(grid, visited, x-1, y);
        //down
        if(x < grid.size() - 1 && grid[x+1][y] == '1')
            dfs(grid, visited, x+1, y);
        //left
        if(y > 0 && grid[x][y-1] == '1')
            dfs(grid, visited, x, y-1);
        //right
        if(y < grid[0].size() - 1 && grid[x][y+1] == '1')
            dfs(grid, visited, x, y+1);
    }
public:
    int numIslands(vector<vector<char>> &grid) {
        int m = grid.size();
        if(m == 0) return 0;
        int n = grid[0].size();
        if(n == 0) return 0;
        vector<vector<bool> > visited(m,vector<bool>(n,false));
        
        int num = 0;
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(grid[i][j] == '1' && !visited[i][j]) {
                    dfs(grid, visited, i, j);
                    num++;
                }
            }
        }
        return num;
    }
};

bfs:

class Solution {
    void bfs(vector<vector<char>> &grid, vector<vector<bool>> &visited, int x, int y) {
        queue<pair<int,int>> q;
        q.push(make_pair(x,y));
        visited[x][y] = true;
        while(!q.empty()) {
            auto i = q.front();
            q.pop();
            x = i.first;
            y = i.second;
            if(x > 0 && grid[x-1][y] == '1' && visited[x-1][y] == false) {
                q.push(make_pair(x-1,y));
                visited[x-1][y] = true;
            }
            if(x < grid.size() - 1 && grid[x+1][y] == '1' && visited[x+1][y] == false) {
                q.push(make_pair(x+1,y));
                visited[x+1][y] = true;
            }
            if(y > 0 && grid[x][y-1] == '1' && visited[x][y-1] == false) {
                q.push(make_pair(x,y-1));
                visited[x][y-1] = true;
            }
            if(y < grid[0].size() - 1 && grid[x][y+1] == '1' && visited[x][y+1] == false) {
                q.push(make_pair(x,y+1));
                visited[x][y+1] = true;
            }
        }
    }
public:
    int numIslands(vector<vector<char>> &grid) {
        int m = grid.size();
        if(m == 0) return 0;
        int n = grid[0].size();
        if(n == 0) return 0;
        vector<vector<bool> > visited(m,vector<bool>(n,false));
        
        int num = 0;
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(grid[i][j] == '1' && !visited[i][j]) {
                    bfs(grid, visited, i, j);
                    num++;
                }
            }
        }
        return num;
    }
};

union-find:

class Solution {
    vector<int> roots;
    int findRoot(int idx) {
        while(roots[idx] != idx)
            idx = roots[idx];
        return idx;
    }
public:
    int numIslands(vector<vector<char>> &grid) {
        int m = grid.size();
        if(m == 0) return 0;
        int n = grid[0].size();
        if(n == 0) return 0;
        
        roots.resize(m * n, -1);
        vector<vector<int>> dirs = {{-1,0},{1,0},{0,-1},{0,1}};
        
        int num = 0;
        for(int i = 0; i < m; i++)
            for(int j = 0; j < n; j++)
                if(grid[i][j] == '1') {
                    roots[i * n + j] = i * n + j;
                    num++;
                }
        cout<<"num="<<num<<endl;
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(grid[i][j] == '0') 
                    continue;
                int id_old = i * n + j;
                for (auto & d:dirs) {
                    if(i + d[0] < 0 || i + d[0] >= m || j + d[1] < 0 || j + d[1] >= n)
                        continue;
                    if(grid[i + d[0]][j + d[1]] == '0')
                        continue;
                    int id_new = (i + d[0]) * n + (j + d[1]);
                    int root_new = findRoot(id_new);
                    int root_old = findRoot(id_old);
                    if(root_new != root_old) {
                        roots[root_new] = root_old;
                        num--;
                    }
                }
            }
        }
        return num;
    }
};

number of island ii:
https://leetcode.com/problems/number-of-islands-ii/

class Solution {
    vector<int> parents;
    int num;
    void uf(int old_id, int new_id) {
        int old_root = find(old_id);
        int new_root = find(new_id);
        if(old_root != new_root) {
            parents[old_root] = new_root;
            num--;
        }
    }
    
    int find(int id) {
        while(parents[id] != id)
            id = parents[id];
        return id;
    }
public:
    vector<int> numIslands2(int m, int n, vector<pair<int, int>>& positions) {
        parents.resize(m * n, -1);
        vector<int> res;
        num = 0;
        for(auto &i : positions) {
            int x = i.first;
            int y = i.second;
            int id = x * n + y;
            parents[id] = id;
            num++;
            if(x > 0 && parents[(x - 1) * n + y] != -1)
                uf(id, (x - 1) * n + y);
            if(x < m - 1 && parents[(x + 1) * n + y] != -1)
                uf(id, (x + 1) * n + y);
            if(y > 0 && parents[x * n + (y - 1)] != -1)
                uf(id, x * n + (y - 1));
            if(y < n - 1 && parents[x * n + (y + 1)] != -1)
                uf(id, x * n + (y + 1));
            res.push_back(num);
        }
        return res;
    }
};    
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 229,732評論 6 539
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 99,214評論 3 426
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 177,781評論 0 382
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,588評論 1 316
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,315評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,699評論 1 327
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,698評論 3 446
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,882評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,441評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,189評論 3 356
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,388評論 1 372
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,933評論 5 363
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,613評論 3 348
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,023評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,310評論 1 293
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,112評論 3 398
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,334評論 2 377

推薦閱讀更多精彩內容