码迷,mamicode.com
首页 > 移动开发 > 详细

小米Git

时间:2017-08-07 11:44:34      阅读:146      评论:0      收藏:0      [点我收藏+]

标签:一个队列   int   个数   字符串   ack   get   tor   题目   初始化   

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树的根节点。)

示例1

输入

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

输出

1

思路:刚开始看题没怎么看懂,后来结合别人的理解才明白。 首先说明一点,题目是求树上任意两点的最近分割点。会想到什么?这是一棵什么树,如果看过剑指offer的题的话,会想到有一道这样的题目:求树中两个节点的最低公共祖先,看起来很像。可是仔细研究后会发现,那里的题目中给的是树的链接形式,而这里给出的是邻接矩阵。在那道题目中,如果二叉树为普通的树,思路是先要分别求出两个节点到根节点的路径,然后将问题转化为两个链表的第一个公共节点,这样问题就容易解决了。所以,我们要考虑将邻接矩阵转化,可以转化为一个父亲节点的数组,每个索引值对应的位置存放着这个节点的父节点的索引值。然后我们可以根据这个父节点数组来进行遍历,找出两个节点到根节点的路径。其中,有一个问题,就是将邻接矩阵转化为父亲节点的数组时,可以用一个队列,类似于层序遍历的思想,将每个节点的父节点都找出来。这样的话,在找出两条路径后,就可以采用求两个链表的第一个公共节点的方法求。

需要注意的一点是,在利用堆栈判断两个路径元素时,一定将判断堆栈是否为空的条件写在前面,写在后面的话,会先判断栈顶元素,导致程序直接崩溃。

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) {
        if (indexA == indexB){    //若是两个节点索引相同,返回其中一个节点
            return indexA; 
        }
        if (indexA * indexB == 0){ // 若有一个节点为根节点,返回根节点
            return 0;
        }
        int n = matrix.size();    // 节点的个数
        vector<int> father(n,0);  // 初始化father数组,存放每个结点的父节点
        vector<int> flag(n);      // 标志向量
        father[0] = -1;           // 根节点的父节点为-1
        flag[0] = 1;              // 根节点已经访问过
        queue<int> children;      // 用来层次遍历整个树
        children.push(0);         // 压入根节点
        
        // 构造father 数组,从根节点开始,利用层次遍历的思想
        while (!children.empty()){
            int parent = children.front(); // 当前的节点
            children.pop();               //  弹出当前节点
            for (int i = 0; i < (matrix[parent].size()); i++){   // 遍历当前节点的下一层
                if (flag[i] != 1 && matrix[parent][i] == 1){ //若与当前节点连接并且还没被访问过
                    father[i] = parent;   // 
                    children.push(i);
                    flag[i] = 1;
                }
            }
        }
        
        int ia = indexA;
        int ib = indexB;
        // 记录从根节点到a, b节点的路径
        stack<int> pathtoA;
        stack<int> pathtoB;
        while (ia != -1){
            pathtoA.push(ia);
            ia = father[ia];
        }
        while (ib != -1){
            pathtoB.push(ib);
            ib = father[ib];
        }
        int tmp = 0;
        while ( !pathtoA.empty() && !pathtoB.empty() && pathtoA.top() == pathtoB.top()){ // 这个判断的条件尤其注意顺序
            tmp = pathtoA.top();
            pathtoA.pop();
            pathtoB.pop();
        }
        
        return tmp;
        
    }
};

 

小米Git

标签:一个队列   int   个数   字符串   ack   get   tor   题目   初始化   

原文地址:http://www.cnblogs.com/simplepaul/p/7297959.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!