1. 程式人生 > 實用技巧 >java 楊輝三角

java 楊輝三角

技術標籤:leetcode C++leetcode

有 n 個城市,其中一些彼此相連,另一些沒有相連。如果城市 a 與城市 b 直接相連,且城市 b 與城市 c 直接相連,那麼城市 a 與城市 c 間接相連。

省份 是一組直接或間接相連的城市,組內不含其他沒有相連的城市。

給你一個 n x n 的矩陣 isConnected ,其中 isConnected[i][j] = 1 表示第 i 個城市和第 j 個城市直接相連,而 isConnected[i][j] = 0 表示二者不直接相連。

返回矩陣中 省份 的數量。

示例 1:


輸入:isConnected = [[1,1,0],[1,1,0],[0,0,1]]

輸出:2
示例 2:


輸入:isConnected = [[1,0,0],[0,1,0],[0,0,1]]
輸出:3

提示:

1 <= n <= 200
n == isConnected.length
n == isConnected[i].length
isConnected[i][j] 為 1 或 0
isConnected[i][i] == 1
isConnected[i][j] == isConnected[j][i]


分析:

深度遍歷或者廣度遍歷


class Solution {
private:
        // 從i開始的深度優先遍歷
        int DFS(vector<vector<int>>& isConnected, int i, int *visit);
        // 找到第i行的第一個鄰居
        int findNeighbor(vector<vector<int>>& isConnected, int i, int *visit);        

public:
    int findCircleNum(vector<vector<int>>& isConnected) {
        //     訪問陣列
        int *visit = new int[isConnected.size()];
        int size = isConnected.size();
        int result = 0;

        // 遍歷每一個座標
        for(int i=0; i<size; i++){
                result += DFS(isConnected,i, visit);
        }
        return result;
    }
};
// 深度優先遍歷
int Solution::DFS(vector<vector<int>>& isConnected, int i, int *visit){
        int nebor=findNeighbor(isConnected, i, visit);
        if(nebor < 0) return 0;
        for(nebor=findNeighbor(isConnected, i, visit); nebor>=0; nebor=findNeighbor(isConnected, i, visit)){
                visit[nebor] = 1;
                DFS(isConnected, nebor, visit);
        }
        return 1;
}
// 找到第i行的第一個鄰居
int Solution::findNeighbor(vector<vector<int>>& isConnected, int i, int *visit){
        int size = isConnected.size();
        for(int j=0; j<size; j++){
                if(isConnected[i][j]>0 && visit[j]<1) return j;
        }
        return -1;
}