小米Git

小米Git

描述

git是一种分布式代码管理工具，git通过树的形式记录文件的更改历史，比如： base’<–base<–A<–A’ ^ | — B<–B’ 小米工程师常常需要寻找两个分支最近的分割点，即base.假设git 树是多叉树，请实现一个算法，计算git树上任意两点的最近分割点。 （假设git树节点数为n,用邻接矩阵的形式表示git树：字符串数组matrix包含n个字符串，每个字符串由字符’0’或’1’组成，长度为n。matrix[i][j]==’1’当且仅当git树种第i个和第j个节点有连接。节点0为git树的根节点。）

样例输入

[01011,10100,01000,10000,10000],1,2

样例输出

1

解决思路

其实这就是一个多叉树求共同父节点的问题，先dfs获取两个节点的路径，然后比较求得最近的路径节点即可

代码

class Solution {
public:
    /**
    * 返回git树上两点的最近分割点
    *
    * @param matrix 接邻矩阵，表示git树，matrix[i][j] == '1' 当且仅当git树中第i个和第j个节点有连接，节点0为git树的跟节点
    * @param indexA 节点A的index
    * @param indexB 节点B的index
    * @return 整型
    */

    int getSplitNode(vector<string> matrix, int indexA, int indexB) {
        vector<bool>visited;
        for (int i = 0; i < matrix.size(); i++) visited.push_back(false);
        vector<int>path;
        path.push_back(0);//先加入根节点
        visited[0] = true;//先遍历根节点
        vector<int> pathA;
        //A点的路径
        getPath(matrix, 0,indexA,visited,path,pathA);
        for (int i = 0; i < matrix.size(); i++) visited[i]=false;
        visited[0] = true;//先遍历根节点
        vector<int> pathB;
        //B点的路径
        getPath(matrix, 0, indexB, visited, path, pathB);
        //查找路径最大相同点
        int samePath;
        for (int i = 0; i < pathA.size(); i++){
            for (int j = 0; j < pathB.size(); j++){
                if (pathA[i] == pathB[j]) {
                    samePath = pathA[i];
                    break;
                }
            }
        }
        return samePath;
    }

    void getPath(vector<string> matrix,int root,int target,vector<bool>visited,vector<int> path,vector<int>&result){
        if (root == target) {
            result = path;
            return;
        }
        for (int i = 0; i < matrix.size(); i++){//依次遍历节点下的每个节点
            if (matrix[root][i] == '1' && !visited[i]){
                visited[i] = true;//访问该节点
                path.push_back(i);//将节点添加进路径
                getPath(matrix, i, target, visited, path,result);//获取下一节点路径
                path.pop_back();//清空节点，恢复
                if (result.size()) break;//找到就退出
            }
        }
    }
｝