小米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树的根节点。）

IDEA

简单来说，就是求给定多叉树中任意两个节点的最近公共父节点。

要用到多叉树（连通图）的层序遍历，遍历过程中要记录每个节点的父节点parent[i]，和每个节点的所在的层高depth[i]

CODE

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) {        int n=matrix.size();    vector<vector<int> > tree(n);        vector<int> parent(n,-1);        vector<int> depth(n,0);                 for(int i=0;i<n;i++){            for(int j=i+1;j<n;j++){                if(matrix[i][j]=='1'){                    tree[i].push_back(j);                    tree[j].push_back(i);                }            }        }        queue<int> que;        que.push(0);        parent[0]=0;        depth[0]=0;        while(!que.empty()){            int cur=que.front();            que.pop();            for(int k=0;k<tree[cur].size();k++){                int v=tree[cur][k];                if(parent[v]==-1){                    parent[v]=cur;                depth[v]=depth[cur]+1;                que.push(v);                }            }        }        int a=indexA;        int b=indexB;        while(depth[a]>depth[b]){            a=parent[a];}          while(depth[a]<depth[b]){            b=parent[b];        }         while(a!=b){            a=parent[a];            b=parent[b];        }        return a;    }};