寻找共同的父节点

2024年6月7日 223次阅读

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

参考了http://blog.csdn.net/shenxuan520/article/details/73832618

解题思路

  1. 从矩阵构造出父节点记录。
  2. 记录从根节点到带求节点A和B的路径。
  3. 比较路径，找到最近的公共节点。
#include <iostream>
#include <string>
#include <vector>
#include <stack>
using namespace std;

int getSplitNode(vector<string> matrix, int indexA, int indexB) {
    stack<int> s;
    if(indexA==indexB)
        return indexA;
    int len=matrix.size();
    vector<int> father(len,-1);//父节点记录表
    vector<int> flag(len,0);// 保证不重复入栈
    s.push(0);//先将根节点入栈
    flag[0] = 1;
    
    int count = 0; 
    int temp;//记录当前栈顶
    while(!s.empty()){
        temp = s.top();
        s.pop();
        for(int count = 1; count<len;count++){
            if(matrix[temp][count] == '1' && flag[count] ==0){
                s.push(count);
                flag[count] = 1;
                father[count] = temp;//记录父节点
            }
        }
    }
    
    
    vector<int> patha;patha.push_back(indexA);
    vector<int> pathb;pathb.push_back(indexB);
    
    int ta = indexA;
    int tb = indexB;
    
    while(father[ta] != -1){
    	patha.push_back(father[ta]);
    	ta = father[ta];
    }
        	
    while(father[tb] != -1){
    	pathb.push_back(father[tb]);
    	tb = father[tb];
    }
        	
    ta = patha.size()-1;
    tb = pathb.size()-1;
    while(ta>= 0 && tb >= 0 && patha[ta] == pathb[tb]){
    	ta--;
    	tb--;
    }
    
    return pathb[tb+1];// or patha[ta+1]
}
    
int main() {
	// your code goes here
	vector<string> matrix;
	matrix.push_back("0001");
	matrix.push_back("0001");
	matrix.push_back("0001");
	matrix.push_back("1110");

	cout<<getSplitNode(matrix,3,2)<<endl;
	return 0;
}

点赞