小米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;
    }
};