小米Git

    xiaoxiao2021-03-25  94

    题目描述

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

    转载请注明原文地址: https://ju.6miu.com/read-26278.html

    最新回复(0)